fixed hunyuan i2v

2025-05-17 19:57:46 +02:00
parent 2dd9260197
commit ca9e9b94cb
1 changed files with 1 additions and 0 deletions
--- 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]