From f6e0e5d5a4312e5d913f3a91507605435efd0fbd Mon Sep 17 00:00:00 2001 From: Prince Canuma Date: Fri, 16 Jan 2026 15:59:22 +0100 Subject: [PATCH] Update av_ca_timestep_scale_multiplier to 1000 in model configuration for consistency across modules --- mlx_video/convert.py | 2 +- mlx_video/models/ltx/config.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mlx_video/convert.py b/mlx_video/convert.py index 1e08e5c..e251fc5 100644 --- a/mlx_video/convert.py +++ b/mlx_video/convert.py @@ -463,7 +463,7 @@ def create_model_from_config(config: Dict[str, Any]) -> LTXModel: positional_embedding_max_pos=config.get("positional_embedding_max_pos", [20, 2048, 2048]), audio_positional_embedding_max_pos=config.get("audio_positional_embedding_max_pos", [20]), timestep_scale_multiplier=config.get("timestep_scale_multiplier", 1000), - av_ca_timestep_scale_multiplier=config.get("av_ca_timestep_scale_multiplier", 1), + av_ca_timestep_scale_multiplier=config.get("av_ca_timestep_scale_multiplier", 1000), norm_eps=config.get("norm_eps", 1e-6), ) diff --git a/mlx_video/models/ltx/config.py b/mlx_video/models/ltx/config.py index 1f45fb1..6ac9de2 100644 --- a/mlx_video/models/ltx/config.py +++ b/mlx_video/models/ltx/config.py @@ -123,7 +123,7 @@ class LTXModelConfig(BaseModelConfig): # Timestep config timestep_scale_multiplier: int = 1000 - av_ca_timestep_scale_multiplier: int = 1 + av_ca_timestep_scale_multiplier: int = 1000 # Normalization norm_eps: float = 1e-6