Code

app.py

@@ -34,6 +34,7 @@ from tqdm import tqdm
 import imageio
 import av
 import uuid
+import time
 
 from pipeline import CausalInferencePipeline
 from demo_utils.constant import ZERO_VAE_CACHE
@@ -293,6 +294,7 @@ def video_generation_handler_streaming(prompt, seed=42, fps=15):
     # Generation loop
     for idx, current_num_frames in enumerate(all_num_frames):
         print(f"📦 Processing block {idx+1}/{num_blocks}")
+        start = time.monotonic()
         
         noisy_input = noise[:, current_start_frame : current_start_frame + current_num_frames]
 
@@ -390,7 +392,8 @@ def video_generation_handler_streaming(prompt, seed=42, fps=15):
                 
                 # Calculate final progress for this block
                 total_progress = (idx + 1) / num_blocks * 100
-                
+                frame_time = time.monotonic() - start
+                print(f"🔍 DEBUG Block {idx} frame time: {frame_time:.2f}s")
                 # Yield the actual video chunk
                 yield ts_path, gr.update()