From 84818b4a154363e3db692512dc77d1fd7c7bf322 Mon Sep 17 00:00:00 2001 From: Tarjei Mandt Date: Fri, 20 Mar 2026 15:13:37 +1100 Subject: [PATCH] Fix SSM dt clamp default for Nemotron-H --- mlx_lm/models/nemotron_h.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mlx_lm/models/nemotron_h.py b/mlx_lm/models/nemotron_h.py index 0c391e42c..353de36c9 100644 --- a/mlx_lm/models/nemotron_h.py +++ b/mlx_lm/models/nemotron_h.py @@ -61,8 +61,8 @@ class ModelArgs(BaseModelArgs): _block_type_to_char = {"mamba": "M", "attention": "*", "moe": "E", "mlp": "-"} def __post_init__(self): - if self.time_step_limit is None and self.time_step_min is not None: - self.time_step_limit = (self.time_step_min, float("inf")) + if self.time_step_limit is None: + self.time_step_limit = (0.0, float("inf")) # Normalize to hybrid_override_pattern (single-char list) if self.hybrid_override_pattern is None and self.layers_block_type is not None: