From d1fa47722b87f2422cdd5c1d293b20b2f6b9c0a9 Mon Sep 17 00:00:00 2001
From: Prince Canuma <prince.gdt@gmail.com>
Date: Wed, 11 Mar 2026 18:30:29 +0100
Subject: [PATCH] Fix timestep_conditioning logic in infer_vae_decoder_config
 to ensure consistent behavior based on has_timestep flag.

---
 mlx_video/models/ltx/convert.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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