Refactor LTX-2 model structure

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

View File

@@ -1,4 +1,4 @@
from mlx_video.models.ltx import LTXModel, LTXModelConfig
from mlx_video.models.ltx_2 import LTXModel, LTXModelConfig
from mlx_video.convert import (
load_transformer_weights,
load_vae_weights,
@@ -9,7 +9,7 @@ from mlx_video.convert import (
)
# Audio VAE components
from mlx_video.models.ltx.audio_vae import (
from mlx_video.models.ltx_2.audio_vae import (
AudioDecoder,
Vocoder,
decode_audio,
@@ -19,7 +19,7 @@ from mlx_video.models.ltx.audio_vae import (
)
# Conditioning
from mlx_video.conditioning import (
from mlx_video.models.ltx_2.conditioning import (
VideoConditionByLatentIndex,
)