Refactor weight loading and utility functions for LTX-2 model; remove deprecated weight loading file and update imports accordingly

This commit is contained in:
Prince Canuma
2026-03-16 22:25:22 +01:00
parent dd573d53d2
commit 7a576bfbf4
5 changed files with 182 additions and 789 deletions

View File

@@ -1633,7 +1633,7 @@ def generate_video(
a2v_sr = None
if is_a2v:
from mlx_video.models.ltx_2.audio_vae.audio_processor import load_audio, ensure_stereo, waveform_to_mel
from mlx_video.convert import convert_audio_encoder
from mlx_video.models.ltx_2.utils import convert_audio_encoder
from mlx_video.models.ltx_2.audio_vae import AudioEncoder
with console.status("[blue]Loading and encoding input audio (A2V)...[/]", spinner="dots"):