fix LTX-2.3 audio

2026-03-15 02:06:35 +01:00
parent eb0d1355e4
commit 53bae534e7
4 changed files with 649 additions and 130 deletions
--- a/mlx_video/models/ltx/audio_vae/init.py
+++ b/mlx_video/models/ltx/audio_vae/init.py
@@ -9,12 +9,13 @@ from .normalization import NormType, PixelNorm, build_normalization_layer
 from .ops import AudioLatentShape, AudioPatchifier, PerChannelStatistics
 from .resnet import LRELU_SLOPE, ResBlock1, ResBlock2, ResnetBlock
 from .upsample import Upsample, build_upsampling_path
-from .vocoder import Vocoder
+from .vocoder import Vocoder, load_vocoder

 __all__ = [
    # Main components
    "AudioDecoder",
    "Vocoder",
+    "load_vocoder",
    "decode_audio",
    # Ops
    "AudioLatentShape",