Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,6 +24,8 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
|
|
| 24 |
"FireRedTeam/FireRed-Image-Edit-1.1",
|
| 25 |
torch_dtype=dtype
|
| 26 |
).to(device)
|
|
|
|
|
|
|
| 27 |
|
| 28 |
prompt_handler = PromptAugment()
|
| 29 |
|
|
|
|
| 24 |
"FireRedTeam/FireRed-Image-Edit-1.1",
|
| 25 |
torch_dtype=dtype
|
| 26 |
).to(device)
|
| 27 |
+
pipe.vae.enable_tiling()
|
| 28 |
+
pipe.vae.enable_slicing()
|
| 29 |
|
| 30 |
prompt_handler = PromptAugment()
|
| 31 |
|