From 7690aa68a53cea4fa14ab8d0983a82c27e0be158 Mon Sep 17 00:00:00 2001
From: sigma <signsur4739379373@users.noreply.huggingface.co>
Date: Mon, 19 Jan 2026 01:27:01 +0000
Subject: [PATCH] Update app.py

---
 app.py | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

diff --git a/app.py b/app.py
index c3a008c..0bb1fba 100644
--- a/app.py
+++ b/app.py
@@ -239,19 +239,20 @@ from huggingface_hub import hf_hub_download
 import torch.nn.functional as F
 
 # --- 1. setup pipeline with lightning (this works fine) ---
-pipe = QwenImageEditPlusPipeline.from_pretrained(
-    "Qwen/Qwen-Image-Edit-2511", 
+pipe = QwenImageEditPlusPipeline.from_single_file(
+    "path/to/Qwen-Rapid-AIO-NSFW-v21.safetensors",
+    original_config="Qwen/Qwen-Image-Edit-2511", # pulls the config from the base repo
     scheduler=scheduler,
-    torch_dtype=dtype
-).to(device)
+    torch_dtype=torch.bfloat16 # use bf16 for speed on zerogpu
+).to("cuda")
 
-print("loading lightning lora...")
-pipe.load_lora_weights(
-    "lightx2v/Qwen-Image-Edit-2511-Lightning", 
-    weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors"
-)
-pipe.fuse_lora()
-print("lightning lora fused.")
+# print("loading lightning lora...")
+# pipe.load_lora_weights(
+#     "lightx2v/Qwen-Image-Edit-2511-Lightning", 
+#     weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors"
+# )
+# pipe.fuse_lora()
+# print("lightning lora fused.")
 
 
 # # Apply the same optimizations from the first version