Update feifeilib/feifeimodload.py

feifeilib/feifeimodload.py

@@ -19,14 +19,16 @@ def feifeimodload():
         "aifeifei798/DarkIdol-flux-v1.1", torch_dtype=dtype, vae=taef1
     ).to(device)
 
-    # pipe = DiffusionPipeline.from_pretrained(
-    #     "shuttleai/shuttle-3.1-aesthetic", torch_dtype=dtype, vae=taef1
-    # ).to(device)    
+    pipe.load_lora_weights(
+        hf_hub_download("aifeifei798/feifei-flux-lora-v1", "feifei.safetensors"),
+        adapter_name="feifei",
+    )
+
+    pipe.load_lora_weights(
+        hf_hub_download("adirik/flux-cinestill", "lora.safetensors"),
+        adapter_name="fluxcinestill",
+    )
     
-    #pipe.load_lora_weights(
-    #    hf_hub_download("aifeifei798/feifei-flux-lora-v1", "feifei.safetensors"),
-    #    adapter_name="feifei",
-    #)
     #pipe.load_lora_weights(
     #    hf_hub_download(
     #        "aifeifei798/feifei-flux-lora-v1", "FLUX-dev-lora-add_details.safetensors"
@@ -44,21 +46,22 @@ def feifeimodload():
     #    ["feifei", "FLUX-dev-lora-add_details", "Shadow-Projection"],
     #    adapter_weights=[0.65, 0.35, 0.35],
     #)
+    
     #pipe.fuse_lora(
     #    adapter_name=["feifei", "FLUX-dev-lora-add_details", "Shadow-Projection"],
     #    lora_scale=1.0,
     #)
 
-    #pipe.set_adapters(
-    #    ["feifei"],
-    #    adapter_weights=[0.9],
-    #)
-    #pipe.fuse_lora(
-    #    adapter_name=["feifei"],
-    #    lora_scale=1.0,
-    #)
-    pipe.load_lora_weights(hf_hub_download("aifeifei798/feifei-flux-lora-v1", "feifei.safetensors"),adapter_name="feifei")
-    pipe.load_lora_weights("adirik/flux-cinestill")
+    pipe.set_adapters(
+        ["feifei","fluxcinestill"],
+        adapter_weights=[1,1],
+    )
+    
+    pipe.fuse_lora(
+        adapter_name=["feifei","fluxcinestill"],
+        lora_scale=1.0,
+    )
+
     pipe.vae.enable_tiling()
     pipe.unload_lora_weights()
     torch.cuda.empty_cache()