fix: fix nightly test from updating transformers (#6658)

ce9d690e · Mick · GitHub · bdaefbbf · ce9d690e
Unverified Commit ce9d690e authored May 27, 2025 by Mick Committed by GitHub May 27, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

python/sglang/srt/configs/model_config.py python/sglang/srt/configs/model_config.py +1 -0

No files found.
--- a/python/sglang/srt/configs/model_config.py
+++ b/python/sglang/srt/configs/model_config.py
@@ -199,6 +199,7 @@ class ModelConfig:
            if (
                "MistralModel" in self.hf_config.architectures
                or "MixtralForCausalLM" in self.hf_config.architectures
+                or "MistralForCausalLM" in self.hf_config.architectures
            ):
                if getattr(self, "head_dim", None) is None:
                    self.head_dim = (