Skip to content

GitLab

"test/old-api/parsertests.cpp" did not exist on "0d41a7de7cb216b908cc14e9e15d17e4e7f077a2"
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):
self.self_attn = LlamaAttention(
hidden_size=self.hidden_size,
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_scaling=rope_scaling,
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