Refactor audio VAE directory structure and update related paths in conversion and loading functions

2026-03-16 21:53:37 +01:00
parent a6a6bb2166
commit dd573d53d2
3 changed files with 15 additions and 9 deletions
--- a/mlx_video/models/ltx_2/generate.py
+++ b/mlx_video/models/ltx_2/generate.py
@@ -1360,7 +1360,7 @@ def load_audio_decoder(model_path: Path, pipeline: PipelineType):
    """Load audio VAE decoder."""
    from mlx_video.models.ltx_2.audio_vae import AudioDecoder

-    decoder = AudioDecoder.from_pretrained(model_path / "audio_vae")
+    decoder = AudioDecoder.from_pretrained(model_path / "audio_vae" / "decoder")

    return decoder