Spaces:
Running on Zero
Running on Zero
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 filesCrash 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>
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,
|