LPX55 commited on
Commit
4985631
·
verified ·
1 Parent(s): cb3b613

Update app_fast.py

Browse files
Files changed (1) hide show
  1. app_fast.py +2 -2
app_fast.py CHANGED
@@ -83,9 +83,9 @@ Please provide the rewritten instruction in a clean `json` format as:
83
  dtype = torch.bfloat16
84
  device = "cuda" if torch.cuda.is_available() else "cpu"
85
  pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype=dtype).to(device)
 
86
  pipe.load_lora_weights(
87
- "lightx2v/Qwen-Image-Edit-Lightning",
88
- weight_name="Qwen-Image-Edit-Lightning-8steps-V1.1.safetensors"
89
  )
90
  pipe.fuse_lora()
91
 
 
83
  dtype = torch.bfloat16
84
  device = "cuda" if torch.cuda.is_available() else "cpu"
85
  pipe = QwenImageEditPipeline.from_pretrained("Qwen/Qwen-Image-Edit", torch_dtype=dtype).to(device)
86
+ # Load LoRA weights for acceleration
87
  pipe.load_lora_weights(
88
+ "lightx2v/Qwen-Image-Lightning", weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
 
89
  )
90
  pipe.fuse_lora()
91