[Cleanup] Move mamba-max-states-per-path validation into _handle_mamba_backend (#32502)

This commit is contained in:
YAMY
2026-07-28 14:21:21 -07:00
committed by GitHub
parent b8b9f3c8f5
commit dd67452b4f
3 changed files with 9 additions and 15 deletions
+6 -10
View File
@@ -3409,8 +3409,6 @@ class ServerArgs:
# _handle_model_specific_adjustments never runs.
self._resolved_overrides = []
self._validate_mamba_max_states_per_path()
if self.model_path.lower() in ["none", "dummy"]:
return
@@ -3588,14 +3586,6 @@ class ServerArgs:
materialize_declarations(self)
def _validate_mamba_max_states_per_path(self):
value = self.mamba_max_states_per_path
if value == 0 or value < -1:
raise ValueError(
"--mamba-max-states-per-path must be -1 (unlimited) or a positive "
f"integer, got {value}."
)
def _handle_model_capability_adjustments(self):
if parse_connector_type(self.model_path) == ConnectorType.INSTANCE:
return
@@ -5812,6 +5802,12 @@ class ServerArgs:
if self.mamba_cache_philox_rounds < 0:
raise ValueError("--mamba-cache-philox-rounds must be non-negative.")
if self.mamba_max_states_per_path == 0 or self.mamba_max_states_per_path < -1:
raise ValueError(
"--mamba-max-states-per-path must be -1 (unlimited) or a positive "
f"integer, got {self.mamba_max_states_per_path}."
)
if self.enable_mamba_cache_stochastic_rounding:
if self.mamba_ssm_dtype != "float16":
raise ValueError(