Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -15,17 +15,9 @@ def feifeimodload():
         device
     )
     pipe = DiffusionPipeline.from_pretrained(
-        "sayakpaul/FLUX.1-merged", torch_dtype=dtype, vae=taef1
+        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
     ).to(device)
 
-    #pipe = DiffusionPipeline.from_pretrained(
-    #    "aifeifei798/shuttle-3-diffusion", torch_dtype=dtype, vae=taef1
-    #).to(device)
-    
-    # pipe = DiffusionPipeline.from_pretrained(
-    #     "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, vae=taef1
-    # ).to(device)
-
     pipe.load_lora_weights(
         hf_hub_download("aifeifei798/feifei-flux-lora-v1", "feifei.safetensors"),
         adapter_name="feifei",