From 72fe241ea6660d79a7b1d7cd937bcb0e01acd8d8 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 24 Jan 2025 17:57:35 +0200 Subject: [PATCH] Update pipeline_hunyuan_video.py --- hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py index 43e7006..f963fcb 100644 --- a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py +++ b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py @@ -291,7 +291,7 @@ class HunyuanVideoPipeline(DiffusionPipeline): if latents is None: latents = noise elif leapfusion_img2vid: - noise[:, :, [0,], :, :] = latents[:, :, [0,], :, :] + noise[:, :, [0,], :, :] = latents[:, :, [0,], :, :].to(noise) latents = noise.to(device) elif denoise_strength < 1.0: latents = latents.to(device)