Spaces:
Runtime error
Runtime error
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 = 14
|
| 357 |
current_start_frame = 0
|
| 358 |
all_num_frames = [pipeline.num_frame_per_block] * num_blocks
|
| 359 |
|