Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -90,7 +90,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 90 |
pipe = QwenImageEditPlusPipeline.from_pretrained(
|
| 91 |
"Qwen/Qwen-Image-Edit-2511",
|
| 92 |
transformer=QwenImageTransformer2DModel.from_pretrained(
|
| 93 |
-
"prithivMLmods/Qwen-Image-Edit-Rapid-AIO-
|
| 94 |
torch_dtype=dtype,
|
| 95 |
device_map='cuda'
|
| 96 |
),
|
|
|
|
| 90 |
pipe = QwenImageEditPlusPipeline.from_pretrained(
|
| 91 |
"Qwen/Qwen-Image-Edit-2511",
|
| 92 |
transformer=QwenImageTransformer2DModel.from_pretrained(
|
| 93 |
+
"prithivMLmods/Qwen-Image-Edit-Rapid-AIO-V4",
|
| 94 |
torch_dtype=dtype,
|
| 95 |
device_map='cuda'
|
| 96 |
),
|