From 7efb654cdb33394f61308840217c0a1983b5768d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jukka=20Sepp=C3=A4nen?= <40791699+kijai@users.noreply.github.com> Date: Fri, 11 Oct 2024 02:47:26 +0300 Subject: [PATCH] Update pyramid_dit_for_video_gen_pipeline.py --- pyramid_dit/pyramid_dit_for_video_gen_pipeline.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py b/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py index 704ef2e..349975a 100644 --- a/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py +++ b/pyramid_dit/pyramid_dit_for_video_gen_pipeline.py @@ -399,7 +399,7 @@ class PyramidDiTForVideoGeneration: input_image_latent = input_image_latent.to(dtype).to(device) generated_latents_list = [input_image_latent] # The generated results - last_generated_latents = input_image_latent + #last_generated_latents = input_image_latent self.dit.to(device) comfy_pbar = ProgressBar(num_units) @@ -456,7 +456,7 @@ class PyramidDiTForVideoGeneration: comfy_pbar.update(1) generated_latents_list.append(intermed_latents[-1]) - last_generated_latents = intermed_latents + #last_generated_latents = intermed_latents generated_latents = torch.cat(generated_latents_list, dim=2) @@ -570,7 +570,7 @@ class PyramidDiTForVideoGeneration: stages = self.stages generated_latents_list = [] # The generated results - last_generated_latents = None + #last_generated_latents = None self.dit.to(device) comfy_pbar = ProgressBar(num_units) @@ -644,8 +644,7 @@ class PyramidDiTForVideoGeneration: ) comfy_pbar.update(1) generated_latents_list.append(intermed_latents[-1]) - last_generated_latents = intermed_latents - self.dit.to('cpu') + #last_generated_latents = intermed_latents generated_latents = torch.cat(generated_latents_list, dim=2)