Fix timestep_conditioning logic in infer_vae_decoder_config to ensure consistent behavior based on has_timestep flag.

2026-03-11 18:30:29 +01:00
parent 207c223354
commit d1fa47722b
1 changed files with 1 additions and 1 deletions
--- a/mlx_video/models/ltx/convert.py
+++ b/mlx_video/models/ltx/convert.py
@@ -494,7 +494,7 @@ def infer_vae_decoder_config(weights: Dict[str, mx.array], variant: str) -> dict
        "num_res_blocks": 2,
        "out_ch": 2,
        "resolution": 256,
-        "timestep_conditioning": has_timestep if has_timestep else (variant == "dev"),
+        "timestep_conditioning": has_timestep,
        "z_channels": 8,
    }
    return config