Qwen-Image-Edit_Fast-Presets

Running on Zero

LPX55 commited on Aug 18

Commit

91627f2

verified ·

1 Parent(s): 0ee077e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -86,12 +86,13 @@ Please provide the rewritten instruction in a clean `json` format as:
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype=dtype).to(device)
 pipe.load_lora_weights(
-    "lightx2v/Qwen-Image-Edit-Lightning",
-    weight_name="Qwen-Image-Edit-Lightning-8steps-V1.1.safetensors"
 )
 pipe.fuse_lora()
 @spaces.GPU(duration=60)
 def infer(
     image,

 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype=dtype).to(device)
+# Load LoRA weights for acceleration
 pipe.load_lora_weights(
+    "lightx2v/Qwen-Image-Lightning", weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
 )
 pipe.fuse_lora()
 @spaces.GPU(duration=60)
 def infer(
     image,