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

Auto-detect timestep_cond from model metadata ()

Browse Source
This commit is contained in:
Prince Canuma
2026-01-16 14:55:50 +01:00
parent a658911f98
commit e1bff927df
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx_video/generate.py
View File

@@ -265,7 +265,7 @@ def generate_video(
vae_decoder = load_vae_decoder(
str(model_path / 'ltx-2-19b-distilled.safetensors'),
timestep_conditioning=True
timestep_conditioning=None # Auto-detect from model metadata
)
latents = upsample_latents(latents, upsampler, vae_decoder.latents_mean, vae_decoder.latents_std)

2
mlx_video/generate_av.py
View File

@@ -427,7 +427,7 @@ def generate_video_with_audio(
vae_decoder = load_vae_decoder(
str(model_path / 'ltx-2-19b-distilled.safetensors'),
timestep_conditioning=True
timestep_conditioning=None # Auto-detect from model metadata
)
video_latents = upsample_latents(video_latents, upsampler, vae_decoder.latents_mean, vae_decoder.latents_std)