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
f46864d6
Unverified
Commit
f46864d6
authored
Apr 08, 2024
by
egortolmachev
Committed by
GitHub
Apr 08, 2024
Browse files
[Bugfix] Added Command-R GPTQ support (#3849)
Co-authored-by:
Egor Tolmachev
<
t333ga@gmail.com
>
parent
b4543c8f
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
0 deletions
+10
-0
vllm/model_executor/models/commandr.py
vllm/model_executor/models/commandr.py
+10
-0
No files found.
vllm/model_executor/models/commandr.py
View file @
f46864d6
...
@@ -349,11 +349,21 @@ class CohereForCausalLM(nn.Module):
...
@@ -349,11 +349,21 @@ class CohereForCausalLM(nn.Module):
if
shard_name
not
in
name
:
if
shard_name
not
in
name
:
continue
continue
name
=
name
.
replace
(
shard_name
,
param_name
)
name
=
name
.
replace
(
shard_name
,
param_name
)
# Skip loading extra bias for GPTQ models.
if
name
.
endswith
(
".bias"
)
and
name
not
in
params_dict
:
continue
param
=
params_dict
[
name
]
param
=
params_dict
[
name
]
weight_loader
=
param
.
weight_loader
weight_loader
=
param
.
weight_loader
weight_loader
(
param
,
loaded_weight
,
shard_id
)
weight_loader
(
param
,
loaded_weight
,
shard_id
)
break
break
else
:
else
:
# lm_head is not used in vllm as it is tied with embed_token.
# To prevent errors, skip loading lm_head.weight.
if
"lm_head.weight"
in
name
:
continue
# Skip loading extra bias for GPTQ models.
if
name
.
endswith
(
".bias"
)
and
name
not
in
params_dict
:
continue
param
=
params_dict
[
name
]
param
=
params_dict
[
name
]
weight_loader
=
getattr
(
param
,
"weight_loader"
,
weight_loader
=
getattr
(
param
,
"weight_loader"
,
default_weight_loader
)
default_weight_loader
)
...
...
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