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

2025-05-27 15:28:11 +08:00
parent bdaefbbfbd
commit ce9d690ef4
1 changed files with 1 additions and 0 deletions
--- 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 = (