ltxv: overwrite existing mask on conditioned frame. (#7845)

This commit overwrites the noise mask on the latent frame that is being
conditioned with keyframe conditioning, setting it to one.
This commit is contained in:
Andrew Kvochko 2025-04-28 20:42:04 +03:00 committed by GitHub
parent 5a50c3c7e5
commit 772b4c5945
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -153,6 +153,15 @@ class LTXVAddGuide:
return node_helpers.conditioning_set_values(cond, {"keyframe_idxs": keyframe_idxs})
def append_keyframe(self, positive, negative, frame_idx, latent_image, noise_mask, guiding_latent, strength, scale_factors):
_, latent_idx = self.get_latent_index(
cond=positive,
latent_length=latent_image.shape[2],
guide_length=guiding_latent.shape[2],
frame_idx=frame_idx,
scale_factors=scale_factors,
)
noise_mask[:, :, latent_idx:latent_idx + guiding_latent.shape[2]] = 1.0
positive = self.add_keyframe_index(positive, frame_idx, guiding_latent, scale_factors)
negative = self.add_keyframe_index(negative, frame_idx, guiding_latent, scale_factors)