From ca9e9b94cbab1fbc68baab492f79b504834cd3b8 Mon Sep 17 00:00:00 2001 From: DeepBeepMeep Date: Sat, 17 May 2025 19:57:46 +0200 Subject: [PATCH] fixed hunyuan i2v --- hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py | 1 + 1 file changed, 1 insertion(+) diff --git a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py index 9261cf1..840d96d 100644 --- a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py +++ b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py @@ -1331,6 +1331,7 @@ class HunyuanVideoPipeline(DiffusionPipeline): # compute the previous noisy sample x_t -> x_t-1 if i2v_mode and i2v_condition_type == "token_replace": + noise_pred = noise_pred.unsqueeze(0) latents = self.scheduler.step( noise_pred[:, :, 1:, :, :], t, latents[:, :, 1:, :, :], **extra_step_kwargs, return_dict=False )[0]