Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -20,7 +20,7 @@ def feifeimodload():
     #).to(device)
 
     pipe = DiffusionPipeline.from_pretrained(
-        "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype
+        "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
     ).to(device)
     
     pipe.load_lora_weights(
@@ -28,18 +28,18 @@ def feifeimodload():
         adapter_name="feifei",
     )
 
-    pipe.load_lora_weights(
-        hf_hub_download("aifeifei798/flux-nsfw-lora", "Ultra-Realistic Nipple Generator_v2.safetensors"),
-        adapter_name="Ultra-Realistic",
-    )
+    #pipe.load_lora_weights(
+    #    hf_hub_download("aifeifei798/flux-nsfw-lora", "Ultra-Realistic Nipple Generator_v2.safetensors"),
+    #    adapter_name="Ultra-Realistic",
+    #)
 
     pipe.set_adapters(
-        ["feifei", "Ultra-Realistic"],
-        adapter_weights=[0.8, 0.85],
+        ["feifei"],
+        adapter_weights=[0.65],
     )
     
     pipe.fuse_lora(
-        adapter_name=["feifei", "Ultra-Realistic"],
+        adapter_name=["feifei"],
         lora_scale=1.0,
     )