testing

pyramid_dit/pyramid_dit_for_video_gen_pipeline.py

@@ -563,8 +563,8 @@ class PyramidDiTForVideoGeneration:
                 # if not generated_latents_list or len(generated_latents_list) == 0:
                 #     raise ValueError("No latent tensors generated - check previous steps ")
 
-                print(f"Number of generated latents: {len(generated_latents_list)}")
-                print(f"Shapes of generated latents: {[x.shape for x in generated_latents_list]}")
+                print(f"know the tensor of generated_latent_list: {generated_latents_list}")
+                print(f"know the len(stages): {len(stages) - 1}")
                 
                 clean_latents_list = self.get_pyramid_latent(torch.cat(generated_latents_list, dim=2), len(stages) - 1)