Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -11,13 +11,6 @@ def feifeimodload():
 
     dtype = torch.bfloat16
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    #taef1 = AutoencoderTiny.from_pretrained("aifeifei798/taef1", torch_dtype=dtype).to(
-    #    device
-    #)
-    
-    #pipe = DiffusionPipeline.from_pretrained(
-    #    "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype, vae=taef1
-    #).to(device)
 
     pipe = DiffusionPipeline.from_pretrained(
         "aifeifei798/DarkIdol-flux-v1", torch_dtype=dtype
@@ -28,23 +21,13 @@ def feifeimodload():
         adapter_name="feifei",
     )
 
-    pipe.load_lora_weights(
-        hf_hub_download("aifeifei798/AWPortraitCN", "AWPortraitCN.safetensors"),
-        adapter_name="AWPortraitCN",
-    )
-    
-    #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", "AWPortraitCN"],
-        adapter_weights=[0.8,0.5],
+        ["feifei"],
+        adapter_weights=[0.8],
     )
     
     pipe.fuse_lora(
-        adapter_name=["feifei", "AWPortraitCN"],
+        adapter_name=["feifei"],
         lora_scale=1.0,
     )