Update pyramid_dit_for_video_gen_pipeline.py

2024-10-12 15:17:52 +03:00
parent 99acdc520d
commit 5a98e101d4
1 changed files with 1 additions and 2 deletions
@@ -391,8 +391,7 @@ class PyramidDiTForVideoGeneration:
        #input_image_tensor = image_transform(input_image).unsqueeze(0).unsqueeze(2)   # [b c 1 h w]
        
        input_image_latent = input_image_latent.to(dtype).to(device)
-        #generated_latents_list = [input_image_latent]    # The generated results
-        generated_latents_list = list(torch.unbind(input_image_latent, dim=0))
+        generated_latents_list = [input_image_latent]    # The generated results
        #last_generated_latents = input_image_latent

        self.dit.to(device)