Update feifeilib/feifeitexttoimg.py

feifeilib/feifeitexttoimg.py

@@ -17,6 +17,8 @@ def feifeitexttoimg(
         sharpened_select=False,
         styles_Radio=["(None)"],
         FooocusExpansion_select=False,
+        nsfw_select=False, 
+        nsfw_slider=0.45,
         seed=random.randint(0, MAX_SEED),
         randomize_seed=False,
         width=896,
@@ -24,20 +26,29 @@ def feifeitexttoimg(
         num_inference_steps=4,
         guidance_scale=3.5,
         num_strength=0.35,
-        num_feifei=0.85,
+        num_feifei=0.45,
         progress=gr.Progress(track_tqdm=True),
 ):
     prompt,generator = feifeiprompt(randomize_seed,seed,prompt,quality_select,styles_Radio,FooocusExpansion_select)
 
-    pipe.set_adapters(
-        ["feifei"],
-        adapter_weights=[num_feifei],
-    )
-    
-    pipe.fuse_lora(
-        adapter_name=["feifei"],
-        lora_scale=1.0,
-    )
+    if nsfw_select:
+        pipe.set_adapters(
+            ["feifei", "sldr_flux_nsfw_v2"],
+            adapter_weights=[num_feifei,nsfw_slider],
+        )
+        pipe.fuse_lora(
+            adapter_name=["feifei", "sldr_flux_nsfw_v2"],
+            lora_scale=1.0,
+        )
+    else:
+        pipe.set_adapters(
+            ["feifei"],
+            adapter_weights=[num_feifei],
+        )
+        pipe.fuse_lora(
+            adapter_name=["feifei"],
+            lora_scale=1.0,
+        )
 
     #pipe.unload_lora_weights()