Update app.py

app.py

@@ -347,21 +347,15 @@ 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)
-
-
-    num_blocks = 9
-    current_start_frame = 0
-    all_num_frames = [pipeline.num_frame_per_block] * num_blocks
+    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]
 
-
+    num_blocks = 7
+    current_start_frame = 0
+    all_num_frames = [pipeline.num_frame_per_block] * num_blocks
     
     total_frames_yielded = 0