Update pipeline_cogvideox.py

kijai · Nov 24, 2024 · c71bca9 · c71bca9
1 parent 8d6e53b
commit c71bca9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pipeline_cogvideox.py b/pipeline_cogvideox.py
@@ -507,6 +507,7 @@ def __call__(
                     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)