From c71bca935029fa8c3e14b4dae903736343310fc9 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 24 Nov 2024 18:34:00 +0200
Subject: [PATCH] Update pipeline_cogvideox.py

---
 pipeline_cogvideox.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pipeline_cogvideox.py b/pipeline_cogvideox.py
index 59269b2..8b3a71d 100644
--- a/pipeline_cogvideox.py
+++ b/pipeline_cogvideox.py
@@ -507,6 +507,7 @@ class CogVideoXPipeline(DiffusionPipeline, CogVideoXLoraLoaderMixin):
                     image_cond_latents = image_cond_latents.repeat(1, latents.shape[1], 1, 1, 1)
             else:
                 logger.info(f"Received {image_cond_latents.shape[1]} image conditioning frames")
+            image_cond_latents = image_cond_latents.to(self.vae_dtype)
 
         # 6. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
         extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)