Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -171,7 +171,7 @@ scheduler_config = {
|
|
| 171 |
scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
|
| 172 |
|
| 173 |
|
| 174 |
-
pipe = QwenImageEditInpaintPipeline.from_pretrained("Qwen/Qwen-Image-Edit
|
| 175 |
pipe.load_lora_weights(
|
| 176 |
"lightx2v/Qwen-Image-Lightning",
|
| 177 |
weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
|
|
|
|
| 171 |
scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
|
| 172 |
|
| 173 |
|
| 174 |
+
pipe = QwenImageEditInpaintPipeline.from_pretrained("Qwen/Qwen-Image-Edit", scheduler=scheduler, torch_dtype=torch.bfloat16).to("cuda")
|
| 175 |
pipe.load_lora_weights(
|
| 176 |
"lightx2v/Qwen-Image-Lightning",
|
| 177 |
weight_name="Qwen-Image-Lightning-8steps-V1.1.safetensors"
|