Adam Oster Claude Opus 4.8 (1M context) commited on
Commit
8a42e93
·
1 Parent(s): 64b2479

use xlarge ZeroGPU (full Blackwell) for VAE-decode headroom

Browse files

Crash is in VAE-decode conv3d only, after the transformer completes, on
every torch version. On the default 'large' size (half a Blackwell, ~48GB)
the large full-res decode allocation fails and PyTorch's OOM path dies in
the forked process's NVML query, surfacing as the nvmlInit assert. Built-in
VAE tiling drops the decoder's timestep conditioning so it's unsafe here;
request the full 96GB Blackwell instead (2x quota cost).

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -140,7 +140,7 @@ def get_duration(prompt, negative_prompt, input_image_filepath, input_video_file
140
  else:
141
  return 60
142
 
143
- @spaces.GPU(duration=get_duration)
144
  def generate(prompt, negative_prompt, input_image_filepath=None, input_video_filepath=None,
145
  height_ui=512, width_ui=704, mode="text-to-video",
146
  duration_ui=2.0,
 
140
  else:
141
  return 60
142
 
143
+ @spaces.GPU(duration=get_duration, size="xlarge")
144
  def generate(prompt, negative_prompt, input_image_filepath=None, input_video_filepath=None,
145
  height_ui=512, width_ui=704, mode="text-to-video",
146
  duration_ui=2.0,