Update src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py

prishajain1 · github-actions[bot] · web-flow · commit c0c56cbc1729 · 2026-05-12T09:11:35.000+05:30
Co-authored-by: github-actions[bot] &lt;41898282+github-actions[bot]@users.noreply.github.com&gt;
diff --git a/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py b/src/maxdiffusion/pipelines/ltx2/ltx2_pipeline.py
@@ -1689,7 +1689,7 @@ def convert_to_vel(lat, x0, sig):
         audio_latents_jax, self.audio_vae.latents_mean.value, self.audio_vae.latents_std.value
     )
 
-    num_mel_bins = self.audio_vae.config.mel_bins
+    num_mel_bins = self.audio_vae.config.mel_bins if self.audio_vae is not None else 128
     latent_mel_bins = num_mel_bins // self.audio_vae_mel_compression_ratio
 
     audio_latents = self._unpack_audio_latents(

Original file line number	Diff line number	Diff line change
`@@ -1689,7 +1689,7 @@ def convert_to_vel(lat, x0, sig):`
`1689`	`1689`	`audio_latents_jax, self.audio_vae.latents_mean.value, self.audio_vae.latents_std.value`
`1690`	`1690`	`)`
`1691`	`1691`
`1692`		`- num_mel_bins = self.audio_vae.config.mel_bins`
	`1692`	`+ num_mel_bins = self.audio_vae.config.mel_bins if self.audio_vae is not None else 128`
`1693`	`1693`	`latent_mel_bins = num_mel_bins // self.audio_vae_mel_compression_ratio`
`1694`	`1694`
`1695`	`1695`	`audio_latents = self._unpack_audio_latents(`