Spaces:
Running on Zero
Running on Zero
Commit ·
e3b1179
1
Parent(s): 939d211
Drop VAE tiling (it fits at normal decode); PiD has no tiling API to apply during upscale
Browse files
app.py
CHANGED
|
@@ -241,14 +241,6 @@ def generate(
|
|
| 241 |
num_inference_steps = int(num_inference_steps)
|
| 242 |
H = W = int(resolution)
|
| 243 |
|
| 244 |
-
# Heavy resolutions push the VAE decode over the 48GB MIG budget — slice + tile.
|
| 245 |
-
# Mutating pipeline.vae here is fork-local (ZeroGPU forks per call) so no cleanup needed.
|
| 246 |
-
if H >= 768:
|
| 247 |
-
try:
|
| 248 |
-
pipeline.vae.enable_tiling()
|
| 249 |
-
pipeline.vae.enable_slicing()
|
| 250 |
-
except Exception as _e:
|
| 251 |
-
print(f"[pid] VAE tiling/slicing not available: {_e}", flush=True)
|
| 252 |
|
| 253 |
# initial: show the live preview, hide the final slider
|
| 254 |
yield gr.update(visible=True, value=None, label="Generating Z-Image…"), gr.update(visible=False, value=None), gr.update(value=seed)
|
|
|
|
| 241 |
num_inference_steps = int(num_inference_steps)
|
| 242 |
H = W = int(resolution)
|
| 243 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 244 |
|
| 245 |
# initial: show the live preview, hide the final slider
|
| 246 |
yield gr.update(visible=True, value=None, label="Generating Z-Image…"), gr.update(visible=False, value=None), gr.update(value=seed)
|