Refactor weight loading and sanitization processes for audio models

2026-01-23 17:31:25 +01:00
parent 2681f75d2f
commit 02bfa228d9
18 changed files with 510 additions and 498 deletions
--- a/mlx_video/models/ltx/audio_vae/causal_conv_2d.py
+++ b/mlx_video/models/ltx/audio_vae/causal_conv_2d.py
@@ -5,7 +5,7 @@ from typing import Tuple, Union
 import mlx.core as mx
 import mlx.nn as nn

-from .causality_axis import CausalityAxis
+from ..config import CausalityAxis


 def _pair(x: Union[int, Tuple[int, int]]) -> Tuple[int, int]: