Update LTX-Video/ltx_video/pipelines/pipeline_ltx_video.py

LTX-Video/ltx_video/pipelines/pipeline_ltx_video.py

@@ -977,7 +977,7 @@ class LTXVideoPipeline(DiffusionPipeline):
         print(f"[2ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
         print(f"skip_initial_inference_steps {skip_initial_inference_steps}")
         print(f"skip_final_inference_steps {skip_final_inference_steps}")
-        #print(f"latents {latents.shape}")
+        print(f"latents {latents.shape}")
 
         if self.allowed_inference_steps is not None:
             for timestep in [round(x, 4) for x in timesteps.tolist()]:
@@ -1121,7 +1121,7 @@ class LTXVideoPipeline(DiffusionPipeline):
         print(f"[5ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
         print(f"skip_initial_inference_steps {skip_initial_inference_steps}")
         print(f"skip_final_inference_steps {skip_final_inference_steps}")
-        #print(f"latents {latents.shape}")
+        print(f"latents {latents.shape}")
         
 
         # Update the latents with the conditioning items and patchify them into (b, n, c)
@@ -1143,7 +1143,7 @@ class LTXVideoPipeline(DiffusionPipeline):
         print(f"[6ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
         print(f"skip_initial_inference_steps {skip_initial_inference_steps}")
         print(f"skip_final_inference_steps {skip_final_inference_steps}")
-        #print(f"latents {latents.shape}")
+        print(f"latents {latents.shape}")
         # 6. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
         extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
 
@@ -1151,7 +1151,7 @@ class LTXVideoPipeline(DiffusionPipeline):
         print(f"[7ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
         print(f"skip_initial_inference_steps {skip_initial_inference_steps}")
         print(f"skip_final_inference_steps {skip_final_inference_steps}")
-        #print(f"latents {latents.shape}")
+        print(f"latents {latents.shape}")
         # 7. Denoising loop
         num_warmup_steps = max(
             len(timesteps) - num_inference_steps * self.scheduler.order, 0
@@ -1347,9 +1347,9 @@ class LTXVideoPipeline(DiffusionPipeline):
         print(f"[8ADUC DEBUG LTX *causal_video_autoencoder.py*]=======")
         print(f"skip_initial_inference_steps {skip_initial_inference_steps}")
         print(f"skip_final_inference_steps {skip_final_inference_steps}")
-        #print(f"latents {latents.shape}")
+        print(f"latents {latents.shape}")
       
-          if offload_to_cpu:
+        if offload_to_cpu:
             self.transformer = self.transformer.cpu()
             if self._execution_device == "cuda":
                 torch.cuda.empty_cache()