LTXV: fix key frame noise mask dimensions for when real noise mask exists

This commit is contained in:
Harel Cain 2025-08-19 14:05:35 +03:00
parent 4977f203fa
commit 47780b10dd

View File

@ -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,