add audio

mlx_video/models/ltx/audio_vae/causality_axis.py

@@ -0,0 +1,12 @@
+"""Causality axis enum for specifying causal convolution dimensions."""
+
+from enum import Enum
+
+
+class CausalityAxis(Enum):
+    """Enum for specifying the causality axis in causal convolutions."""
+
+    NONE = None
+    WIDTH = "width"
+    HEIGHT = "height"
+    WIDTH_COMPATIBILITY = "width-compatibility"