n0p/mlx-video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor LTX-2 model structure

Browse Source
This commit is contained in:
Prince Canuma
2026-03-16 14:50:01 +01:00
parent decb3eb9e5
commit 3a0da19adb
50 changed files with 3882 additions and 3365 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_rope.py
View File

@@ -2,10 +2,10 @@ import pytest
import mlx.core as mx
import numpy as np
from mlx_video.models.ltx.rope import (
from mlx_video.models.ltx_2.rope import (
precompute_freqs_cis,
)
from mlx_video.models.ltx.config import LTXModelConfig, LTXRopeType
from mlx_video.models.ltx_2.config import LTXModelConfig, LTXRopeType
def create_video_position_grid(

4
tests/test_vae_streaming.py
View File

@@ -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,