Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -12,13 +12,13 @@ def feifeimodload():
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
     
-    #pipe = DiffusionPipeline.from_pretrained(
-    #    "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
-    #).to(device)
-
     pipe = DiffusionPipeline.from_pretrained(
-        "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
+        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
     ).to(device)
+
+    # pipe = DiffusionPipeline.from_pretrained(
+    #     "black-forest-labs/FLUX.1-schnell", torch_dtype=dtype
+    # ).to(device)
     
     pipe.load_lora_weights(
         hf_hub_download("aifeifei798/feifei-flux-lora-v1.1", "feifei-v1.1.safetensors"),
@@ -27,7 +27,7 @@ def feifeimodload():
 
     pipe.set_adapters(
         ["feifei"],
-        adapter_weights=[0.75],
+        adapter_weights=[0.85],
     )
     
     pipe.fuse_lora(