Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -353,7 +353,7 @@ def video_generation_handler_streaming(prompt, seed=42, fps=15, save_frames=True
|
|
353 |
if not APP_STATE["current_use_taehv"] and not args.trt:
|
354 |
vae_cache = [c.to(device=gpu, dtype=torch.float16) for c in ZERO_VAE_CACHE]
|
355 |
|
356 |
-
num_blocks =
|
357 |
current_start_frame = 0
|
358 |
all_num_frames = [pipeline.num_frame_per_block] * num_blocks
|
359 |
|
|
|
353 |
if not APP_STATE["current_use_taehv"] and not args.trt:
|
354 |
vae_cache = [c.to(device=gpu, dtype=torch.float16) for c in ZERO_VAE_CACHE]
|
355 |
|
356 |
+
num_blocks = 9
|
357 |
current_start_frame = 0
|
358 |
all_num_frames = [pipeline.num_frame_per_block] * num_blocks
|
359 |
|