Update pipeline_hunyuan_video.py

2025-01-24 17:57:35 +02:00
parent d4f102b7e8
commit 72fe241ea6
1 changed files with 1 additions and 1 deletions
@@ -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)