Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -2,26 +2,18 @@ import torch
 import spaces
 from diffusers import (
     DiffusionPipeline,
-    AutoencoderTiny,
-    FluxControlNetModel
+    AutoencoderTiny
 )
 from huggingface_hub import hf_hub_download
-from diffusers.utils import load_image
-from diffusers.pipelines import FluxControlNetPipeline
 
 def feifeimodload():
 
     dtype = torch.bfloat16
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
-    # Load pipeline
-    controlnet = FluxControlNetModel.from_pretrained(
-      "jasperai/Flux.1-dev-Controlnet-Upscaler",
-      torch_dtype=torch.bfloat16
-    )
     
-    pipe = FluxControlNetPipeline.from_pretrained(
-        "aifeifei798/DarkIdol-flux-v1", controlnet=controlnet, torch_dtype=dtype, 
+    pipe = DiffusionPipeline.from_pretrained(
+        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
     ).to(device)
     
     pipe.load_lora_weights(