debugging

optimization.py

@@ -60,6 +60,7 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
         quantize_(pipeline.transformer, Float8DynamicActivationFloat8WeightConfig())
 
         hidden_states: torch.Tensor = call.kwargs["hidden_states"]
+        print(f"{hidden_states.shape=}")
         unpacked_hidden_states = LTXConditionPipeline._unpack_latents(
             hidden_states,
             latent_num_frames,
@@ -68,6 +69,7 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
             TRANSFORMER_SPATIAL_PATCH_SIZE,
             TRANSFORMER_TEMPORAL_PATCH_SIZE,
         )
+        print(f"{unpacked_hidden_states.shape=}")
         unpacked_hidden_states_transposed = unpacked_hidden_states.transpose(-1, -2).contiguous()
         if unpacked_hidden_states.shape[-1] > hidden_states.shape[-2]:
             hidden_states_landscape = unpacked_hidden_states