up

optimization.py

@@ -84,6 +84,7 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
         hidden_states_portrait = LTXConditionPipeline._pack_latents(
             hidden_states_portrait, TRANSFORMER_SPATIAL_PATCH_SIZE, TRANSFORMER_TEMPORAL_PATCH_SIZE
         )
+        print(f"{hidden_states_landscape.shape=}, {hidden_states_portrait.shape=}")
 
         exported_landscape = torch.export.export(
             mod=pipeline.transformer,