From fe01885acf892de636b1b2743903812099bd42e3 Mon Sep 17 00:00:00 2001
From: Harel Cain <harel@lightricks.com>
Date: Wed, 20 Aug 2025 09:33:10 +0200
Subject: [PATCH] LTXV: fix key frame noise mask dimensions for when real noise
 mask exists (#9425)

---
 comfy_extras/nodes_lt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy_extras/nodes_lt.py b/comfy_extras/nodes_lt.py
index b5058667a..f82337a67 100644
--- a/comfy_extras/nodes_lt.py
+++ b/comfy_extras/nodes_lt.py
@@ -166,7 +166,7 @@ class LTXVAddGuide:
         negative = self.add_keyframe_index(negative, frame_idx, guiding_latent, scale_factors)
 
         mask = torch.full(
-            (noise_mask.shape[0], 1, guiding_latent.shape[2], 1, 1),
+            (noise_mask.shape[0], 1, guiding_latent.shape[2], noise_mask.shape[3], noise_mask.shape[4]),
             1.0 - strength,
             dtype=noise_mask.dtype,
             device=noise_mask.device,