diff --git a/python/sglang/srt/configs/model_config.py b/python/sglang/srt/configs/model_config.py
index 96a558266..7a90cf413 100644
--- 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 = (