apolinario commited on
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
Files changed (1) hide show
  1. app.py +0 -8
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)