fix: ltxv crop guides works with 0 keyframes (#7085)

This patch fixes a bug in LTXVCropGuides when the latent has no keyframes. Additionally, the first frame is always added as a keyframe. Co-authored-by: Andrew Kvochko <[email protected]>
comfyanonymous · Mar 5, 2025 · 872780d · 872780d
1 parent 6d45ffb
commit 872780d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/comfy_extras/nodes_lt.py b/comfy_extras/nodes_lt.py
@@ -194,11 +194,6 @@ def generate(self, positive, negative, vae, latent, image, frame_idx, strength):
         frame_idx, latent_idx = self.get_latent_index(positive, latent_length, frame_idx, scale_factors)
         assert latent_idx + t.shape[2] <= latent_length, "Conditioning frames exceed the length of the latent sequence."
 
-        if frame_idx == 0:
-            latent_image, noise_mask = self.replace_latent_frames(latent_image, noise_mask, t, latent_idx, strength)
-            return (positive, negative, {"samples": latent_image, "noise_mask": noise_mask},)
-
-
         num_prefix_frames = min(self._num_prefix_frames, t.shape[2])
 
         positive, negative, latent_image, noise_mask = self.append_keyframe(
@@ -252,6 +247,8 @@ def crop(self, positive, negative, latent):
         noise_mask = get_noise_mask(latent)
 
         _, num_keyframes = get_keyframe_idxs(positive)
+        if num_keyframes == 0:
+            return (positive, negative, {"samples": latent_image, "noise_mask": noise_mask},)
 
         latent_image = latent_image[:, :, :-num_keyframes]
         noise_mask = noise_mask[:, :, :-num_keyframes]