n0p/mlx-video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update av_ca_timestep_scale_multiplier to 1000 in model configuration for consistency across modules

Browse Source
This commit is contained in:
Prince Canuma
2026-01-16 15:59:22 +01:00
parent e1bff927df
commit f6e0e5d5a4
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx_video/convert.py
View File

@@ -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]), 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]), audio_positional_embedding_max_pos=config.get("audio_positional_embedding_max_pos", [20]),
timestep_scale_multiplier=config.get("timestep_scale_multiplier", 1000), 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), norm_eps=config.get("norm_eps", 1e-6),
) )

2
mlx_video/models/ltx/config.py
View File

@@ -123,7 +123,7 @@ class LTXModelConfig(BaseModelConfig):
# Timestep config # Timestep config
timestep_scale_multiplier: int = 1000 timestep_scale_multiplier: int = 1000
av_ca_timestep_scale_multiplier: int = 1 av_ca_timestep_scale_multiplier: int = 1000
# Normalization # Normalization
norm_eps: float = 1e-6 norm_eps: float = 1e-6