Skip to content

GitLab

Unverified Commit 0c48b37c authored by Philipp Moritz's avatar Philipp Moritz Committed by GitHub
Browse files

Fix internlm after https://github.com/vllm-project/vllm/pull/2860 (#2861)

parent 7eacffd9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
vllm/model_executor/models/llama.py
View file @ 0c48b37c
...@@ -175,7 +175,8 @@ class LlamaDecoderLayer(nn.Module): ...@@ -175,7 +175,8 @@ class LlamaDecoderLayer(nn.Module):
self.self_attn = LlamaAttention( self.self_attn = LlamaAttention(
hidden_size=self.hidden_size, hidden_size=self.hidden_size,
num_heads=config.num_attention_heads, num_heads=config.num_attention_heads,
num_kv_heads=config.num_key_value_heads, num_kv_heads=getattr(config, "num_key_value_heads",
config.num_attention_heads),
rope_theta=rope_theta, rope_theta=rope_theta,
rope_scaling=rope_scaling, rope_scaling=rope_scaling,
max_position_embeddings=max_position_embeddings, max_position_embeddings=max_position_embeddings,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment