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
cb944d58
Unverified
Commit
cb944d58
authored
Feb 12, 2025
by
Daniel Han
Committed by
GitHub
Feb 12, 2025
Browse files
Allow Unsloth Dynamic 4bit BnB quants to work (#12974)
parent
d46d490c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
2 deletions
+10
-2
vllm/model_executor/layers/quantization/bitsandbytes.py
vllm/model_executor/layers/quantization/bitsandbytes.py
+10
-2
No files found.
vllm/model_executor/layers/quantization/bitsandbytes.py
View file @
cb944d58
...
...
@@ -133,9 +133,17 @@ def is_layer_skipped_bnb(prefix: str, llm_int8_skip_modules: List[str]):
components
=
prefix
.
split
(
'.'
)
# Check if any of the skip modules exactly matches any component
return
any
(
module_name
in
components
substr_check
=
any
(
module_name
in
components
for
module_name
in
llm_int8_skip_modules
)
# Allow certain layers to not be quantized
set_components
=
set
(
"."
.
join
(
components
[:
i
+
1
])
for
i
in
range
(
len
(
components
)))
set_llm_int8_skip_modules
=
set
(
llm_int8_skip_modules
)
prefix_check
=
len
(
set_llm_int8_skip_modules
&
set_components
)
!=
0
return
substr_check
or
prefix_check
class
BitsAndBytesLinearMethod
(
LinearMethodBase
):
"""Linear method for BitsAndBytes.
...
...
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