Update app.py

app.py

@@ -347,8 +347,11 @@ def video_generation_handler_streaming(prompt, seed=42, fps=15, save_frames=True
     rnd = torch.Generator(gpu).manual_seed(int(seed))
     pipeline._initialize_kv_cache(1, torch.float16, device=gpu)
     pipeline._initialize_crossattn_cache(1, torch.float16, device=gpu)
-    noise = torch.randn([1, 21, 16, 60, 104], device=gpu, dtype=torch.float16, generator=rnd)
     
+    # Calculate total frames needed for your desired number of blocks
+    total_frames = num_blocks * pipeline.num_frame_per_block
+    noise = torch.randn([1, total_frames, 16, 60, 104], device=gpu, dtype=torch.float16, generator=rnd)
+
     vae_cache, latents_cache = None, None
     if not APP_STATE["current_use_taehv"] and not args.trt:
         vae_cache = [c.to(device=gpu, dtype=torch.float16) for c in ZERO_VAE_CACHE]