added timing

Author: Serenagu525

src/maxdiffusion/configs/ltx_video.yml

@@ -24,7 +24,7 @@ pipeline_type: multi-scale
 prompt: "A woman with light skin, wearing a blue jacket and a black hat with a veil, looks down and to her right, then back up as she speaks; she has brown hair styled in an updo, light brown eyebrows, and is wearing a white collared shirt under her jacket; the camera remains stationary on her face as she speaks; the background is out of focus, but shows trees and people in period clothing; the scene is captured in real-life footage."
 height: 512
 width: 512
-num_frames: 344 #344
+num_frames: 88 #344
 flow_shift: 5.0
 fps: 24
 downscale_factor: 0.6666666

src/maxdiffusion/generate_ltx_video.py

@@ -11,6 +11,7 @@
 
 import os
 import torch
+import time
 from pathlib import Path
 
 
@@ -93,6 +94,7 @@ def run(config):
   pipeline = LTXVideoPipeline.from_pretrained(config, enhance_prompt=enhance_prompt)
   if config.pipeline_type == "multi-scale": 
     pipeline = LTXMultiScalePipeline(pipeline)
+  s0 = time.perf_counter()
   images = pipeline(
       height=height_padded,
       width=width_padded,
@@ -102,6 +104,19 @@ def run(config):
       config=config,
       enhance_prompt = False
   )
+  print("compile time: ", (time.perf_counter() - s0))
+  s0 = time.perf_counter()
+  images = pipeline(
+      height=height_padded,
+      width=width_padded,
+      num_frames=num_frames_padded,
+      is_video=True,
+      output_type="pt",
+      config=config,
+      enhance_prompt = False
+  )
+  print("generation time: ", (time.perf_counter() - s0))
+  
   (pad_left, pad_right, pad_top, pad_bottom) = padding
   pad_bottom = -pad_bottom
   pad_right = -pad_right

src/maxdiffusion/pipelines/ltx_video/ltx_video_pipeline.py

@@ -561,7 +561,6 @@ def __call__(
       skip_block_list: Optional[Union[List[List[int]], List[int]]] = None,
       **kwargs,
   ):
-    import pdb; pdb.set_trace()
     prompt = self.config.prompt
     is_video = kwargs.get("is_video", False)
     if prompt is not None and isinstance(prompt, str):