Update feifeilib/feifeitexttoimg.py

feifeilib/feifeitexttoimg.py

@@ -24,14 +24,30 @@ def feifeitexttoimg(
         num_inference_steps=4,
         guidance_scale=3.5,
         num_strength=0.35,
+        FeiFei=0.85,
         progress=gr.Progress(track_tqdm=True),
 ):
     prompt,generator = feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
+
+    pipe.load_lora_weights(
+        hf_hub_download("aifeifei798/feifei-flux-lora-v1.1", "feifei-v1.1.safetensors"),
+        adapter_name="feifei",
+    )
+
+    pipe.set_adapters(
+        ["feifei"],
+        adapter_weights=[FeiFei],
+    )
+    
+    pipe.fuse_lora(
+        adapter_name=["feifei"],
+        lora_scale=1.0,
+    )
     
     image = pipe(
         # prompt="",
         # prompt_2=prompt,
-        prompt=prompt,
+        prompt_2=prompt,
         width=width,
         height=height,
         num_inference_steps=num_inference_steps,