From fcc0f3e65a78c983a045a939943a90026a4feef4 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 20 Dec 2024 17:00:42 +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 7e43413..17702f1 100644
--- a/pipeline_cogvideox.py
+++ b/pipeline_cogvideox.py
@@ -515,6 +515,7 @@ class CogVideoXPipeline(DiffusionPipeline, CogVideoXLoraLoaderMixin):
                     image_cond_latents = torch.cat([first_frame, image_cond_latents], dim=1)
                     fun_mask_first_frame = fun_mask[:, : image_cond_frame_count % patch_size_t, ...]
                     fun_mask = torch.cat([fun_mask_first_frame, fun_mask], dim=1)
+                    fun_mask[:, 1:, ...] = 0
             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