Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -3,17 +3,25 @@ import spaces
 from diffusers import (
     DiffusionPipeline,
     AutoencoderTiny,
+    FluxControlNetModel
 )
 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 = DiffusionPipeline.from_pretrained(
-        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
+        "aifeifei798/DarkIdol-flux-v1", controlnet=controlnet, torch_dtype=dtype, 
     ).to(device)
     
     pipe.load_lora_weights(