From 5a98e101d460ca6aed0c9a5c8defdfed31262800 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 12 Oct 2024 15:17:52 +0300
Subject: [PATCH] Update pyramid_dit_for_video_gen_pipeline.py

---
 pyramid_dit/pyramid_dit_for_video_gen_pipeline.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py b/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py
index f8ada58..7708804 100644
--- a/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py
+++ b/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py
@@ -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)