Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
bc0c0192
Unverified
Commit
bc0c0192
authored
Apr 09, 2024
by
Kiran R
Committed by
GitHub
Apr 08, 2024
Browse files
[Bugfix] Enable Proper `attention_bias` Usage in Llama Model Configuration (#3767)
Co-authored-by:
roy
<
jasonailu87@gmail.com
>
parent
f46864d6
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
1 deletion
+5
-1
vllm/model_executor/models/llama.py
vllm/model_executor/models/llama.py
+5
-1
No files found.
vllm/model_executor/models/llama.py
View file @
bc0c0192
...
@@ -184,6 +184,10 @@ class LlamaDecoderLayer(nn.Module):
...
@@ -184,6 +184,10 @@ class LlamaDecoderLayer(nn.Module):
max_position_embeddings
=
getattr
(
config
,
"max_position_embeddings"
,
max_position_embeddings
=
getattr
(
config
,
"max_position_embeddings"
,
8192
)
8192
)
sliding_window
=
getattr
(
config
,
"sliding_window"
,
None
)
sliding_window
=
getattr
(
config
,
"sliding_window"
,
None
)
# Support abacusai/Smaug-72B-v0.1 with attention_bias
# Support internlm/internlm-7b with bias
attention_bias
=
getattr
(
config
,
"attention_bias"
,
False
)
or
getattr
(
config
,
"bias"
,
False
)
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
,
...
@@ -193,7 +197,7 @@ class LlamaDecoderLayer(nn.Module):
...
@@ -193,7 +197,7 @@ class LlamaDecoderLayer(nn.Module):
rope_scaling
=
rope_scaling
,
rope_scaling
=
rope_scaling
,
max_position_embeddings
=
max_position_embeddings
,
max_position_embeddings
=
max_position_embeddings
,
linear_method
=
linear_method
,
linear_method
=
linear_method
,
bias
=
getattr
(
config
,
"bias"
,
False
)
,
bias
=
attention_bias
,
sliding_window
=
sliding_window
,
sliding_window
=
sliding_window
,
)
)
self
.
mlp
=
LlamaMLP
(
self
.
mlp
=
LlamaMLP
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment