From 9848de79463a08611d0a8acbe03f744e84276dc1 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Wed, 14 Jan 2026 17:27:01 -0700
Subject: [PATCH] Fix issue with ltx cached latents if there is no audio.

---
 extensions_built_in/diffusion_models/ltx2/ltx2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extensions_built_in/diffusion_models/ltx2/ltx2.py b/extensions_built_in/diffusion_models/ltx2/ltx2.py
index 6e50d3bc..f6b544a8 100644
--- a/extensions_built_in/diffusion_models/ltx2/ltx2.py
+++ b/extensions_built_in/diffusion_models/ltx2/ltx2.py
@@ -838,7 +838,7 @@ class LTX2Model(BaseModel):
                     batch_size,
                     num_channels_latents=num_channels_latents_audio,
                     num_mel_bins=num_mel_bins,
-                    num_frames=batch.tensor.shape[1],
+                    num_frames=batch.dataset_config.num_frames,
                     frame_rate=frame_rate,
                     sampling_rate=self.pipeline.audio_sampling_rate,
                     hop_length=self.pipeline.audio_hop_length,