Refactor LoRA loading logic in app.py to use lazy loading

app.py

@@ -129,18 +129,13 @@ LORA_CONFIGS = {
     "Meta4": ("loras/Meta4.safetensors", 0.85),
     "Oily_Skin_QWEN_V2_GMR": ("loras/Oily Skin QWEN V2-GMR.safetensors", 0.85),
     "PillowHump_2509": ("loras/PillowHump_2509.safetensors", 0.85),
-    "Put_it_here_Qwen_edit_V2_0": ("loras/Put it here_Qwen edit_V2.0.safetensors", 0.85),
     "QWEN_jtn_barbell": ("loras/QWEN_jtn_barbell.safetensors", 0.85),
     "Qwen_Image_Helm_v0_1": ("loras/Qwen-Image-Helm_v0.1.safetensors", 0.85),
     "Qwen_MysticXXX_v1": ("loras/Qwen-MysticXXX-v1.safetensors", 0.85),
-    "Qwen_NSFW_Beta2": ("loras/Qwen-NSFW-Beta2.safetensors", 0.85),
-    "Qwen_NSFW_Beta4": ("loras/Qwen-NSFW-Beta4.safetensors", 0.85),
     "Qwen_NSFW_Beta5": ("loras/Qwen-NSFW-Beta5.safetensors", 0.85),
-    "Qwen_NSFW": ("loras/Qwen-NSFW.safetensors", 0.85),
     "Qwen_Real_PS_v1_83K": ("loras/Qwen-Real PS_v1_83K.safetensors", 0.85),
     "Qwen4Play_v2": ("loras/Qwen4Play_v2.safetensors", 0.85),
     "QwenImageHentaiPIV_v3_1": ("loras/QwenImageHentaiPIV_v3.1.safetensors", 0.85),
-    "QwenSnofs1_1": ("loras/QwenSnofs1_1.safetensors", 0.85),
     "Qwen_Nsfw_Body_V10_4K": ("loras/Qwen_Nsfw_Body_V10-4K.safetensors", 0.85),
     "Qwen_Nsfw_Body_V14_10K": ("loras/Qwen_Nsfw_Body_V14-10K.safetensors", 0.85),
     "Qwen_Real_Nud3s": ("loras/Qwen_Real_Nud3s.safetensors", 0.85),
@@ -173,7 +168,6 @@ LORA_CONFIGS = {
     "qwen_hand_grab_6000s": ("loras/qwen_hand_grab_6000s.safetensors", 0.85),
     "qwen_image_edit_remove_clothing_v1_0": ("loras/qwen_image_edit_remove-clothing_v1.0.safetensors", 0.85),
     "qwen_image_snapchat": ("loras/qwen_image_snapchat.safetensors", 0.85),
-    "qwen_snofs": ("loras/qwen_snofs.safetensors", 0.85),
     "qwen_uncensor_000014928": ("loras/qwen_uncensor_000014928.safetensors", 0.85),
     "reclining_nude_v1_000003500": ("loras/reclining_nude_v1_000003500.safetensors", 0.85),
     "royal_treatment_V3": ("loras/royal+treatment+V3.safetensors", 0.85),
@@ -183,12 +177,8 @@ LORA_CONFIGS = {
     "qwen_real_consistency_A2R_2509_Base": ("loras-2/qwen真实+一致性-A2R_2509_Base.safetensors", 0.6),
 }
 
-# Load all LoRAs
-for adapter_name, (weight_path, _) in LORA_CONFIGS.items():
-    pipe.load_lora_weights("wiikoo/Qwen-lora-nsfw",
-                         weight_name=weight_path,
-                         adapter_name=adapter_name)
-
+# Track which LoRAs have been loaded (lazy loading on first use)
+loaded_adapters = set()
 
 pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 MAX_SEED = np.iinfo(np.int32).max
@@ -228,10 +218,28 @@ def infer(
     if input_image is None:
         raise gr.Error("Please upload an image to edit.")
 
-    # Special case for consistence_edit_v2 which uses two adapters
+    # Lazy load LoRA on first use
     if lora_adapter == "consistence_edit_v2":
+        # Special case: needs two adapters
+        if "consistence_edit_v2" not in loaded_adapters:
+            pipe.load_lora_weights("wiikoo/Qwen-lora-nsfw",
+                                 weight_name="loras2/consistence_edit_v2.safetensors",
+                                 adapter_name="consistence_edit_v2")
+            loaded_adapters.add("consistence_edit_v2")
+        if "qwen_real_consistency_A2R_2509_Base" not in loaded_adapters:
+            pipe.load_lora_weights("wiikoo/Qwen-lora-nsfw",
+                                 weight_name="loras-2/qwen真实+一致性-A2R_2509_Base.safetensors",
+                                 adapter_name="qwen_real_consistency_A2R_2509_Base")
+            loaded_adapters.add("qwen_real_consistency_A2R_2509_Base")
         pipe.set_adapters(["consistence_edit_v2", "qwen_real_consistency_A2R_2509_Base"], adapter_weights=[0.4, 0.6])
     elif lora_adapter in LORA_CONFIGS:
+        # Load LoRA if not already loaded
+        if lora_adapter not in loaded_adapters:
+            weight_path, _ = LORA_CONFIGS[lora_adapter]
+            pipe.load_lora_weights("wiikoo/Qwen-lora-nsfw",
+                                 weight_name=weight_path,
+                                 adapter_name=lora_adapter)
+            loaded_adapters.add(lora_adapter)
         _, weight = LORA_CONFIGS[lora_adapter]
         pipe.set_adapters([lora_adapter], adapter_weights=[weight])
     else: