Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
app.py
CHANGED
|
@@ -101,7 +101,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 101 |
pipe = QwenImageEditPlusPipeline.from_pretrained(
|
| 102 |
"Qwen/Qwen-Image-Edit-2509",
|
| 103 |
transformer=QwenImageTransformer2DModel.from_pretrained(
|
| 104 |
-
"linoyts/Qwen-Image-Edit-Rapid-AIO",
|
| 105 |
subfolder='transformer',
|
| 106 |
torch_dtype=dtype,
|
| 107 |
device_map='cuda'
|
|
|
|
| 101 |
pipe = QwenImageEditPlusPipeline.from_pretrained(
|
| 102 |
"Qwen/Qwen-Image-Edit-2509",
|
| 103 |
transformer=QwenImageTransformer2DModel.from_pretrained(
|
| 104 |
+
"linoyts/Qwen-Image-Edit-Rapid-AIO", # [transformer weights extracted from: Phr00t/Qwen-Image-Edit-Rapid-AIO]
|
| 105 |
subfolder='transformer',
|
| 106 |
torch_dtype=dtype,
|
| 107 |
device_map='cuda'
|