Auto-detect timestep_cond from model metadata ()

2026-01-16 14:55:50 +01:00
parent a658911f98
commit e1bff927df
2 changed files with 2 additions and 2 deletions
--- a/mlx_video/generate.py
+++ b/mlx_video/generate.py
@@ -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)