diff --git a/mlx_video/models/ltx/convert.py b/mlx_video/models/ltx/convert.py index 330c11c..eb4c532 100644 --- 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