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
51ff216d
Unverified
Commit
51ff216d
authored
Dec 21, 2024
by
George
Committed by
GitHub
Dec 21, 2024
Browse files
[Bugfix] update should_ignore_layer (#11354)
Signed-off-by:
George Ohashi
<
george@neuralmagic.com
>
parent
dd2b5633
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
vllm/model_executor/layers/quantization/compressed_tensors/utils.py
..._executor/layers/quantization/compressed_tensors/utils.py
+1
-1
No files found.
vllm/model_executor/layers/quantization/compressed_tensors/utils.py
View file @
51ff216d
...
@@ -30,7 +30,7 @@ def should_ignore_layer(layer_name: Optional[str],
...
@@ -30,7 +30,7 @@ def should_ignore_layer(layer_name: Optional[str],
# in the safetensors checkpoint. So, we convert the name
# in the safetensors checkpoint. So, we convert the name
# from the fused version to unfused + check to make sure that
# from the fused version to unfused + check to make sure that
# each shard of the fused layer has the same scheme.
# each shard of the fused layer has the same scheme.
if
proj_name
in
FUSED_LAYER_NAME_MAPPING
:
if
proj_name
in
FUSED_LAYER_NAME_MAPPING
and
layer_name
not
in
ignore
:
shard_proj_names
=
FUSED_LAYER_NAME_MAPPING
[
proj_name
]
shard_proj_names
=
FUSED_LAYER_NAME_MAPPING
[
proj_name
]
# Convert fused_name --> [shard_names]
# Convert fused_name --> [shard_names]
...
...
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