tsi-org commited on
Commit
b1c2087
·
verified ·
1 Parent(s): da94340

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 14
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