Qwen-Image-Edit-2509-LoRAs-Fast2

Paused

App Files Files Community

Onise commited on 29 days ago

Commit

38b7feb

verified ·

1 Parent(s): e944e94

Update app.py

Browse files

Files changed (1) hide show

app.py +145 -129

app.py CHANGED Viewed

@@ -52,8 +52,7 @@ steel_blue_theme = SteelBlueTheme()
 # ── Device / dtype ─────────────────────────────────────────────────────────────
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-dtype  = torch.bfloat16
 print("CUDA available:", torch.cuda.is_available())
 print("Using device:", device)
@@ -75,8 +74,8 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
 # ── OOM FIX: Enable VAE tiling and slicing to bound VRAM usage ─────────────────
 pipe.vae.enable_tiling(tile_sample_min_width=256, tile_sample_min_height=256)
 pipe.vae.enable_slicing()
-# ───────────────────────────────────────────────────────────────────────────────
 try:
     pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
     print("Flash Attention 3 Processor set successfully.")
@@ -85,127 +84,130 @@ except Exception as e:
 # ── LoRA catalog ──────────────────────────────────────────────────────────
 LORA_REPO = "wiikoo/Qwen-lora-nsfw"
 LORA_CONFIGS = {
-    "CockQwen_v3":              "loras/CockQwen-v3.safetensors",
-    "Eva_Qwen_V3":              "loras/Eva_Qwen_V3.safetensors",
-    "Facial_Cumshots_V1":       "loras/Facial_Cumshots_For_Qwen_Image_V1.safetensors",
-    "HearmemanAI_V3_Breasts":   "loras/HearmemanAI_V3_Rank64_BreastsLoRA_Epoch60.safetensors",
-    "HearmemanAI_V4_Breasts":   "loras/HearmemanAI_V4_Rank128_BreastsLoRA_Epoch80.safetensors",
-    "InniePussy":               "loras/InniePussy.safetensors",
-    "JTT2_5":                   "loras/[QWEN] JTT2_5.safetensors",
-    "LumiNude01a":              "loras/LumiNude01a_CE_QWEN_AIT3k.safetensors",
-    "MEXX_QWEN_TG300":          "loras/MEXX_QWEN_TG300_23.safetensors",
-    "Meta4":                    "loras/Meta4.safetensors",
-    "MysticXXX":                "loras/Qwen-MysticXXX-v1.safetensors",
-    "Nsfw_Body_V10":            "loras/Qwen_Nsfw_Body_V10-4K.safetensors",
-    "Nsfw_Body_V14":            "loras/Qwen_Nsfw_Body_V14-10K.safetensors",
-    "OilySkin_V2":              "loras/Oily Skin QWEN V2-GMR.safetensors",
-    "PillowHump_2509":          "loras/PillowHump_2509.safetensors",
-    "PutItHere_V2":             "loras/Put it here_Qwen edit_V2.0.safetensors",
-    "PutItHere_V01":            "loras/put it here_QwenEdit_V0.1.safetensors",
-    "Qwen4Play_v2":             "loras/Qwen4Play_v2.safetensors",
-    "QwenHentai_v3":            "loras/QwenImageHentaiPIV_v3.1.safetensors",
-    "Qwen_Helm":                "loras/Qwen-Image-Helm_v0.1.safetensors",
-    "Qwen_NSFW_Beta1":          "loras/Qwen-NSFW.safetensors",
-    "Qwen_NSFW_Beta2":          "loras/Qwen-NSFW-Beta2.safetensors",
-    "Qwen_NSFW_Beta4":          "loras/Qwen-NSFW-Beta4.safetensors",
-    "Qwen_NSFW_Beta5":          "loras/Qwen-NSFW-Beta5.safetensors",
-    "Qwen_Real_Nud3s":          "loras/Qwen_Real_Nud3s.safetensors",
-    "Qwen_Real_PS":             "loras/Qwen-Real PS_v1_83K.safetensors",
-    "QwenSnofs_v1":             "loras/qwen_snofs.safetensors",
-    "QwenSnofs_v1_1":           "loras/QwenSnofs1_1.safetensors",
-    "Real_Breast_Nipples":      "loras/Real Breast Nipples-QWEN-[rbn]-GMR.safetensors",
-    "SendDudes":                "loras/[QWEN] SendDudes.safetensors",
-    "SendNudesLite":            "loras/SendNudesLite (Qwen).safetensors",
-    "SendNudesPro_Beta":        "loras/[QWEN] Send Nudes Pro - Beta v1.safetensors",
-    "Ultimate_Breast_Nipples":  "loras/Ultimate Realistic Breast NIPPLES-QWEN-[rab]-GMR.safetensors",
-    "ass_up_QWEN":              "loras/ass_up_QWEN.safetensors",
-    "barbell_nipples_QWEN":     "loras/QWEN_jtn_barbell.safetensors",
-    "bfs_v2_face":              "loras-sfw/face_swap_5500_qwen_image_edit_2509_v1.safetensors",
-    "bfs_v2_focus_face":        "loras-sfw/bfs_v2_000005000.safetensors",
-    "bfs_v2_head":              "loras-sfw/bfs_v2_head_000007000.safetensors",
-    "big_nipples_QWEN":         "loras/big_nipples_QWEN.safetensors",
-    "bumpynipples":             "loras/bumpynipples1.safetensors",
-    "cmslt_cum_on_her":         "loras/cmslt_2509_2.safetensors",
-    "consistence_edit_v1":      "loras-2/consistence_edit_v1.safetensors",
-    "consistence_edit_v2":      "loras2/consistence_edit_v2.safetensors",
-    "d33p7hroa7":               "loras/d33p7hroa7_qwen.safetensors",
-    "d1ck_p3n1s_V1_1":          "loras/qwen-image_d!ck_P3N1S_LoRA_V1.1.safetensors",
-    "goblin_anal_v1":           "loras/goblin_anal_v1_qwen.safetensors",
-    "horseshoe_nipple_rings":   "loras/horseshoe_nipple_rings_QWEN.safetensors",
-    "jib_nudity_fixer":         "loras/jib_qwen_fix_000002750.safetensors",
-    "jillin":                   "loras/jillin1.safetensors",
-    "male_nude":                "loras/lora_nudenan_v1.safetensors",
-    "milk_juggs":               "loras/milk_juggs_QWEN.safetensors",
-    "n00d_b":                   "loras/n00d-b-qwen.safetensors",
-    "nsfw_adv_v1":              "loras/qwen-image_nsfw_adv_v1.0.safetensors",
-    "p0ssy_lora_v1":            "loras/p0ssy_lora_v1.safetensors",
-    "p3nis":                    "loras/p3nis.safetensors",
-    "qwen_MCNL":                "loras/qwen_MCNL_v1.0.safetensors",
-    "qwen_PENISLORA":           "loras/qwen-PENISLORA.safetensors",
-    "qwen_hand_grab":           "loras/qwen_hand_grab_6000s.safetensors",
-    "qwen_uncensor":            "loras/qwen_uncensor_000014928.safetensors",
-    "reclining_nude":           "loras/reclining_nude_v1_000003500.safetensors",
-    "remove_clothing":          "loras/qwen_image_edit_remove-clothing_v1.0.safetensors",
-    "royal_treatment_V3":       "loras/royal+treatment+V3.safetensors",
-    "sabi_character":           "loras-2/sabi_character_v1.safetensors",
-    "snapchat_selfie":          "loras/qwen_image_snapchat.safetensors",
-    "uka_qwen":                 "loras/uka_1_qwen.safetensors",
-    "ultimate_realistic_breast":"loras/ultimate realistic breast.safetensors",
 }
 LORA_TRIGGER_WORDS = {
-    "Qwen4Play_v2":             "d0gg13, c0wg1rl, r3v3rs3_c0wg1rl, m15510n4ry, bl0wj0b, penis",
-    "qwen_MCNL":                "nsfw, cum_on_face, blowjob, cowgirlout, creamp1e, penis, l1ck, missionary, nipples, reversecowgirlpov, vagina",
-    "remove_clothing":          "remove her clothing",
-    "Qwen_Real_Nud3s":          "nud3",
-    "HearmemanAI_V4_Breasts":   "large breasts, hard nipples, erect nipples",
-    "HearmemanAI_V3_Breasts":   "large breasts, hard nipples, erect nipples",
-    "Ultimate_Breast_Nipples":  "rab",
-    "ass_up_QWEN":              "ass up showing pussy and anus",
-    "PillowHump_2509":          "Pillow, Humping",
-    "InniePussy":               "Innie pussy, Clean shaven, Vertical slit",
-    "p0ssy_lora_v1":            "Nude",
-    "CockQwen_v3":              "Erect Penis",
-    "p3nis":                    "holding a p3nis",
-    "qwen_PENISLORA":           "PENISLORA",
-    "Facial_Cumshots_V1":       "cum",
-    "bfs_v2_head":              "head swap, transfer head from image 1 to image 2",
-    "bfs_v2_face":              "keep the face consistent, preserve facial identity",
-    "bfs_v2_focus_face":        "head swap from Image 1 to Image 2",
-    "goblin_anal_v1":           "anal penetration, spread ass",
-    "d33p7hroa7":               "deepthroat, penis deep in mouth",
-    "QwenHentai_v3":            "nsfw, anime style, explicit",
-    "Eva_Qwen_V3":              "Eva_gothic, in a kneeling position",
-    "JTT2_5":                   "massive breasts, large breasts, medium breasts, small breasts",
-    "MEXX_QWEN_TG300":          "nsfw, female body",
-    "OilySkin_V2":              "oilski",
-    "barbell_nipples_QWEN":     "barbell nipple piercings",
-    "Qwen_Helm":                "nsfw, anime style",
-    "MysticXXX":                "nsfw",
-    "Qwen_NSFW_Beta1":          "nsfw",
-    "Qwen_NSFW_Beta2":          "nsfw",
-    "Qwen_NSFW_Beta4":          "nsfw",
-    "Qwen_NSFW_Beta5":          "nsfw",
-    "QwenSnofs_v1":             "sex, missionary, cum, cowgirl, reverse cowgirl, selfie, snapchat selfie, prone position, spooning position, undressing",
-    "QwenSnofs_v1_1":           "nsfw, nude, sex, blowjob, cum, selfie",
-    "Nsfw_Body_V10":            "Hourglass figure, Hairless pussy, Hairly pussy",
-    "Nsfw_Body_V14":            "SSS Waistline, Hairless pussy, Hairly pussy",
-    "SendNudesLite":            "nude",
-    "SendNudesPro_Beta":        "flat chest, small breasts, medium breasts, large breasts, massive breasts, big nipples",
-    "SendDudes":                "Penis",
-    "cmslt_cum_on_her":         "Put cum on her",
-    "horseshoe_nipple_rings":   "horseshoe-ring nipple piercings, circular-barbell nipple piercings",
-    "jib_nudity_fixer":         "nude, nipples, vagina",
-    "jillin":                   "masturbating",
-    "male_nude":                "nudeman",
-    "n00d_b":                   "nude, art photography",
-    "nsfw_adv_v1":              "nsfw",
-    "d1ck_p3n1s_V1_1":          "P3N1S, penis",
-    "qwen_uncensor":            "nsfw, cum_on_face, blowjob, cowgirlout, creamp1e, penis, l1ck, missionary, nipples, reversecowgirlpov, vagina",
-    "royal_treatment_V3":       "lick ass, blowjob",
-    "snapchat_selfie":          "selfie, snapchat",
-    "ultimate_realistic_breast":"urb, realistic breast",
 }
 # Tracks which adapter names have been successfully loaded into the pipeline.
@@ -246,11 +248,24 @@ def load_and_apply_stack(extra_adapters: list[str], extra_weights: list[float]):
         if name not in LOADED_ADAPTERS:
             try:
                 print(f"--- Loading adapter: {name} ---")
-                pipe.load_lora_weights(
-                    LORA_REPO,
-                    weight_name=LORA_CONFIGS[name],
-                    adapter_name=name,
-                )
                 LOADED_ADAPTERS.add(name)
             except Exception as e:
                 # If it already exists from a previous call in this Space session,
@@ -269,7 +284,6 @@ def load_and_apply_stack(extra_adapters: list[str], extra_weights: list[float]):
         pipe.set_adapters(loaded, adapter_weights=weights_out)
     else:
         pipe.disable_lora()
     return loaded, weights_out
@@ -316,7 +330,7 @@ def infer(
     # ── Validate aspect ratio ────────────────────────────────────────────────
     # Extreme ratios (>4:1) produce degenerate latent shapes that crash the
-    # transformer or produce garbage.  Reject early with a clear message.
     image = input_image.convert("RGB")
     w, h = image.size
     ratio = max(w, h) / max(min(w, h), 1)
@@ -375,6 +389,7 @@ def infer(
         torch.cuda.empty_cache()
         # ─────────────────────────────────────────────────────────────────────
 # ── UI ─────────────────────────────────────────────────────────────────────────
 css = """
 #col-container { margin: 0 auto; max-width: 980px; }
@@ -404,6 +419,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
             "**Fast local LoRA-powered image editing** powered by `Qwen-Image-Edit-2509` + `Qwen-Image-Edit-Rapid-AIO-V4` + 50+ community LoRAs.<br>"
             "Upload an image, describe the edit, optionally stack LoRAs, and hit **Edit Image**."
         )
         with gr.Accordion("💡 Quick Tips", open=False):
             gr.Markdown(
                 "- **LoRAs auto-add trigger words** to your prompt when selected.<br>"
@@ -442,6 +458,7 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
                     with gr.Row():
                         gr.Markdown("**Stack up to 6 LoRAs.** Select a LoRA to auto-fill trigger words in the prompt.")
                         clear_btn = gr.Button("🧹 Clear All", size="sm", variant="secondary", scale=0)
                     lora_stack = []
                     for i in range(6):
                         with gr.Row():
@@ -503,7 +520,6 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
         inputs=[output_image],
         outputs=[input_image],
     )
     clear_output_btn.click(
         fn=lambda: gr.update(value=None),
         inputs=None,
@@ -513,4 +529,4 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(
         mcp_server=True, ssr_mode=False, show_error=True
-    )

 # ── Device / dtype ─────────────────────────────────────────────────────────────
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+dtype = torch.bfloat16
 print("CUDA available:", torch.cuda.is_available())
 print("Using device:", device)
 # ── OOM FIX: Enable VAE tiling and slicing to bound VRAM usage ─────────────────
 pipe.vae.enable_tiling(tile_sample_min_width=256, tile_sample_min_height=256)
 pipe.vae.enable_slicing()
+# ───────────────────────────────────────────────────────────────────────────────
 try:
     pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
     print("Flash Attention 3 Processor set successfully.")
 # ── LoRA catalog ──────────────────────────────────────────────────────────
 LORA_REPO = "wiikoo/Qwen-lora-nsfw"
 LORA_CONFIGS = {
+    "CockQwen_v3": "loras/CockQwen-v3.safetensors",
+    "Eva_Qwen_V3": "loras/Eva_Qwen_V3.safetensors",
+    "Facial_Cumshots_V1": "loras/Facial_Cumshots_For_Qwen_Image_V1.safetensors",
+    "HearmemanAI_V3_Breasts": "loras/HearmemanAI_V3_Rank64_BreastsLoRA_Epoch60.safetensors",
+    "HearmemanAI_V4_Breasts": "loras/HearmemanAI_V4_Rank128_BreastsLoRA_Epoch80.safetensors",
+    "InniePussy": "loras/InniePussy.safetensors",
+    "JTT2_5": "loras/[QWEN] JTT2_5.safetensors",
+    "LumiNude01a": "loras/LumiNude01a_CE_QWEN_AIT3k.safetensors",
+    "MEXX_QWEN_TG300": "loras/MEXX_QWEN_TG300_23.safetensors",
+    "Meta4": "loras/Meta4.safetensors",
+    "MysticXXX": "loras/Qwen-MysticXXX-v1.safetensors",
+    "Nsfw_Body_V10": "loras/Qwen_Nsfw_Body_V10-4K.safetensors",
+    "Nsfw_Body_V14": "loras/Qwen_Nsfw_Body_V14-10K.safetensors",
+    "OilySkin_V2": "loras/Oily Skin QWEN V2-GMR.safetensors",
+    "PillowHump_2509": "loras/PillowHump_2509.safetensors",
+    "PutItHere_V2": "loras/Put it here_Qwen edit_V2.0.safetensors",
+    "PutItHere_V01": "loras/put it here_QwenEdit_V0.1.safetensors",
+    "Qwen4Play_v2": "loras/Qwen4Play_v2.safetensors",
+    "QwenHentai_v3": "loras/QwenImageHentaiPIV_v3.1.safetensors",
+    "Qwen_Helm": "loras/Qwen-Image-Helm_v0.1.safetensors",
+    "Qwen_NSFW_Beta1": "loras/Qwen-NSFW.safetensors",
+    "Qwen_NSFW_Beta2": "loras/Qwen-NSFW-Beta2.safetensors",
+    "Qwen_NSFW_Beta4": "loras/Qwen-NSFW-Beta4.safetensors",
+    "Qwen_NSFW_Beta5": "loras/Qwen-NSFW-Beta5.safetensors",
+    "Qwen_Real_Nud3s": "loras/Qwen_Real_Nud3s.safetensors",
+    "Qwen_Real_PS": "loras/Qwen-Real PS_v1_83K.safetensors",
+    "QwenSnofs_v1": "loras/qwen_snofs.safetensors",
+    "QwenSnofs_v1_1": "loras/QwenSnofs1_1.safetensors",
+    "QwenSnofs_1.3": ("Cassanovason69/Qwensnofs", "Qwen_Snofs_1_3.safetensors"),  # ← from other repo
+    "Real_Breast_Nipples": "loras/Real Breast Nipples-QWEN-[rbn]-GMR.safetensors",
+    "SendDudes": "loras/[QWEN] SendDudes.safetensors",
+    "SendNudesLite": "loras/SendNudesLite (Qwen).safetensors",
+    "SendNudesPro_Beta": "loras/[QWEN] Send Nudes Pro - Beta v1.safetensors",
+    "Ultimate_Breast_Nipples": "loras/Ultimate Realistic Breast NIPPLES-QWEN-[rab]-GMR.safetensors",
+    "ass_up_QWEN": "loras/ass_up_QWEN.safetensors",
+    "barbell_nipples_QWEN": "loras/QWEN_jtn_barbell.safetensors",
+    "bfs_v2_face": "loras-sfw/face_swap_5500_qwen_image_edit_2509_v1.safetensors",
+    "bfs_v2_focus_face": "loras-sfw/bfs_v2_000005000.safetensors",
+    "bfs_v2_head": "loras-sfw/bfs_v2_head_000007000.safetensors",
+    "big_nipples_QWEN": "loras/big_nipples_QWEN.safetensors",
+    "bumpynipples": "loras/bumpynipples1.safetensors",
+    "cmslt_cum_on_her": "loras/cmslt_2509_2.safetensors",
+    "consistence_edit_v1": "loras-2/consistence_edit_v1.safetensors",
+    "consistence_edit_v2": "loras2/consistence_edit_v2.safetensors",
+    "d33p7hroa7": "loras/d33p7hroa7_qwen.safetensors",
+    "d1ck_p3n1s_V1_1": "loras/qwen-image_d!ck_P3N1S_LoRA_V1.1.safetensors",
+    "goblin_anal_v1": "loras/goblin_anal_v1_qwen.safetensors",
+    "horseshoe_nipple_rings": "loras/horseshoe_nipple_rings_QWEN.safetensors",
+    "jib_nudity_fixer": "loras/jib_qwen_fix_000002750.safetensors",
+    "jillin": "loras/jillin1.safetensors",
+    "male_nude": "loras/lora_nudenan_v1.safetensors",
+    "milk_juggs": "loras/milk_juggs_QWEN.safetensors",
+    "n00d_b": "loras/n00d-b-qwen.safetensors",
+    "nsfw_adv_v1": "loras/qwen-image_nsfw_adv_v1.0.safetensors",
+    "p0ssy_lora_v1": "loras/p0ssy_lora_v1.safetensors",
+    "p3nis": "loras/p3nis.safetensors",
+    "qwen_MCNL": "loras/qwen_MCNL_v1.0.safetensors",
+    "qwen_PENISLORA": "loras/qwen-PENISLORA.safetensors",
+    "qwen_hand_grab": "loras/qwen_hand_grab_6000s.safetensors",
+    "qwen_uncensor": "loras/qwen_uncensor_000014928.safetensors",
+    "reclining_nude": "loras/reclining_nude_v1_000003500.safetensors",
+    "remove_clothing": "loras/qwen_image_edit_remove-clothing_v1.0.safetensors",
+    "royal_treatment_V3": "loras/royal+treatment+V3.safetensors",
+    "sabi_character": "loras-2/sabi_character_v1.safetensors",
+    "snapchat_selfie": "loras/qwen_image_snapchat.safetensors",
+    "uka_qwen": "loras/uka_1_qwen.safetensors",
+    "ultimate_realistic_breast": "loras/ultimate realistic breast.safetensors",
 }
 LORA_TRIGGER_WORDS = {
+    "Qwen4Play_v2": "d0gg13, c0wg1rl, r3v3rs3_c0wg1rl, m15510n4ry, bl0wj0b, penis",
+    "qwen_MCNL": "nsfw, cum_on_face, blowjob, cowgirlout, creamp1e, penis, l1ck, missionary, nipples, reversecowgirlpov, vagina",
+    "remove_clothing": "remove her clothing",
+    "Qwen_Real_Nud3s": "nud3",
+    "HearmemanAI_V4_Breasts": "large breasts, hard nipples, erect nipples",
+    "HearmemanAI_V3_Breasts": "large breasts, hard nipples, erect nipples",
+    "Ultimate_Breast_Nipples": "rab",
+    "ass_up_QWEN": "ass up showing pussy and anus",
+    "PillowHump_2509": "Pillow, Humping",
+    "InniePussy": "Innie pussy, Clean shaven, Vertical slit",
+    "p0ssy_lora_v1": "Nude",
+    "CockQwen_v3": "Erect Penis",
+    "p3nis": "holding a p3nis",
+    "qwen_PENISLORA": "PENISLORA",
+    "Facial_Cumshots_V1": "cum",
+    "bfs_v2_head": "head swap, transfer head from image 1 to image 2",
+    "bfs_v2_face": "keep the face consistent, preserve facial identity",
+    "bfs_v2_focus_face": "head swap from Image 1 to Image 2",
+    "goblin_anal_v1": "anal penetration, spread ass",
+    "d33p7hroa7": "deepthroat, penis deep in mouth",
+    "QwenHentai_v3": "nsfw, anime style, explicit",
+    "Eva_Qwen_V3": "Eva_gothic, in a kneeling position",
+    "JTT2_5": "massive breasts, large breasts, medium breasts, small breasts",
+    "MEXX_QWEN_TG300": "nsfw, female body",
+    "OilySkin_V2": "oilski",
+    "barbell_nipples_QWEN": "barbell nipple piercings",
+    "Qwen_Helm": "nsfw, anime style",
+    "MysticXXX": "nsfw",
+    "Qwen_NSFW_Beta1": "nsfw",
+    "Qwen_NSFW_Beta2": "nsfw",
+    "Qwen_NSFW_Beta4": "nsfw",
+    "Qwen_NSFW_Beta5": "nsfw",
+    "QwenSnofs_v1": "sex, missionary, cum, cowgirl, reverse cowgirl, selfie, snapchat selfie, prone position, spooning position, undressing",
+    "QwenSnofs_v1_1": "nsfw, nude, sex, blowjob, cum, selfie",
+    "QwenSnofs_1.3": "nsfw, nude, sex, blowjob, cum, selfie",  # same style as previous versions
+    "Nsfw_Body_V10": "Hourglass figure, Hairless pussy, Hairly pussy",
+    "Nsfw_Body_V14": "SSS Waistline, Hairless pussy, Hairly pussy",
+    "SendNudesLite": "nude",
+    "SendNudesPro_Beta": "flat chest, small breasts, medium breasts, large breasts, massive breasts, big nipples",
+    "SendDudes": "Penis",
+    "cmslt_cum_on_her": "Put cum on her",
+    "horseshoe_nipple_rings": "horseshoe-ring nipple piercings, circular-barbell nipple piercings",
+    "jib_nudity_fixer": "nude, nipples, vagina",
+    "jillin": "masturbating",
+    "male_nude": "nudeman",
+    "n00d_b": "nude, art photography",
+    "nsfw_adv_v1": "nsfw",
+    "d1ck_p3n1s_V1_1": "P3N1S, penis",
+    "qwen_uncensor": "nsfw, cum_on_face, blowjob, cowgirlout, creamp1e, penis, l1ck, missionary, nipples, reversecowgirlpov, vagina",
+    "royal_treatment_V3": "lick ass, blowjob",
+    "snapchat_selfie": "selfie, snapchat",
+    "ultimate_realistic_breast": "urb, realistic breast",
 }
 # Tracks which adapter names have been successfully loaded into the pipeline.
         if name not in LOADED_ADAPTERS:
             try:
                 print(f"--- Loading adapter: {name} ---")
+                cfg = LORA_CONFIGS[name]
+                if isinstance(cfg, tuple):
+                    # LoRA from a different Hugging Face repo
+                    repo_id, weight_name = cfg
+                    pipe.load_lora_weights(
+                        repo_id,
+                        weight_name=weight_name,
+                        adapter_name=name,
+                    )
+                else:
+                    # Normal LoRA from the main LORA_REPO
+                    pipe.load_lora_weights(
+                        LORA_REPO,
+                        weight_name=cfg,
+                        adapter_name=name,
+                    )
                 LOADED_ADAPTERS.add(name)
             except Exception as e:
                 # If it already exists from a previous call in this Space session,
         pipe.set_adapters(loaded, adapter_weights=weights_out)
     else:
         pipe.disable_lora()
     return loaded, weights_out
     # ── Validate aspect ratio ────────────────────────────────────────────────
     # Extreme ratios (>4:1) produce degenerate latent shapes that crash the
+    # transformer or produce garbage. Reject early with a clear message.
     image = input_image.convert("RGB")
     w, h = image.size
     ratio = max(w, h) / max(min(w, h), 1)
         torch.cuda.empty_cache()
         # ─────────────────────────────────────────────────────────────────────
 # ── UI ─────────────────────────────────────────────────────────────────────────
 css = """
 #col-container { margin: 0 auto; max-width: 980px; }
             "**Fast local LoRA-powered image editing** powered by `Qwen-Image-Edit-2509` + `Qwen-Image-Edit-Rapid-AIO-V4` + 50+ community LoRAs.<br>"
             "Upload an image, describe the edit, optionally stack LoRAs, and hit **Edit Image**."
         )
         with gr.Accordion("💡 Quick Tips", open=False):
             gr.Markdown(
                 "- **LoRAs auto-add trigger words** to your prompt when selected.<br>"
                     with gr.Row():
                         gr.Markdown("**Stack up to 6 LoRAs.** Select a LoRA to auto-fill trigger words in the prompt.")
                         clear_btn = gr.Button("🧹 Clear All", size="sm", variant="secondary", scale=0)
                     lora_stack = []
                     for i in range(6):
                         with gr.Row():
         inputs=[output_image],
         outputs=[input_image],
     )
     clear_output_btn.click(
         fn=lambda: gr.update(value=None),
         inputs=None,
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(
         mcp_server=True, ssr_mode=False, show_error=True
+    )