Use hidden_size_per_head as head_size fallback (vllm-project#24221)

Signed-off-by: nopperl <[email protected]>

Author: nopperl

vllm/config/__init__.py

@@ -1426,6 +1426,11 @@ def get_head_size(self) -> int:
         if getattr(self.hf_text_config, "head_dim", None) is not None:
             return self.hf_text_config.head_dim
 
+        # NOTE: Some models (such as PLaMo2.1) use `hidden_size_per_head`
+        if getattr(self.hf_text_config, "hidden_size_per_head",
+                   None) is not None:
+            return self.hf_text_config.hidden_size_per_head
+
         # FIXME(woosuk): This may not be true for all models.
         return (self.hf_text_config.hidden_size //
                 self.hf_text_config.num_attention_heads)