Refactor LTX-2 model structure

2026-03-16 14:50:01 +01:00
parent decb3eb9e5
commit 3a0da19adb
50 changed files with 3882 additions and 3365 deletions
--- a/tests/test_vae_streaming.py
+++ b/tests/test_vae_streaming.py
@@ -4,8 +4,8 @@ import pytest
 import mlx.core as mx
 import numpy as np

-from mlx_video.models.ltx.video_vae.sampling import DepthToSpaceUpsample
-from mlx_video.models.ltx.video_vae.tiling import (
+from mlx_video.models.ltx_2.video_vae.sampling import DepthToSpaceUpsample
+from mlx_video.models.ltx_2.video_vae.tiling import (
    TilingConfig,
    compute_trapezoidal_mask_1d,
    decode_with_tiling,