From 029636d2a79986afa74577d1a633a4e370bd6206 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 10 Dec 2024 09:41:35 +0200
Subject: [PATCH] Update pipeline_hunyuan_video.py

---
 hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
index 7d1f154..c7cf3f3 100644
--- a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
+++ b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
@@ -178,7 +178,6 @@ class HunyuanVideoPipeline(DiffusionPipeline):
         height,
         width,
         video_length,
-        dtype,
         device,
         timesteps,
         generator,
@@ -197,7 +196,7 @@ class HunyuanVideoPipeline(DiffusionPipeline):
                 f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
                 f" size of {batch_size}. Make sure the batch size matches the length of the generators."
             )
-        noise = randn_tensor(shape, generator=generator, device=device, dtype=dtype)
+        noise = randn_tensor(shape, generator=generator, device=device, dtype=self.base_dtype)
         if latents is None:
             latents = noise
         else:
@@ -493,7 +492,6 @@ class HunyuanVideoPipeline(DiffusionPipeline):
             height,
             width,
             latent_video_length,
-            prompt_embeds.dtype,
             device,
             timesteps,
             generator,