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
6dd5e528
Unverified
Commit
6dd5e528
authored
Feb 03, 2025
by
Kyle Sayers
Committed by
GitHub
Feb 03, 2025
Browse files
Squelch MLA warning for Compressed-Tensors Models (#12704)
Signed-off-by:
Kyle Sayers
<
kylesayrs@gmail.com
>
parent
c11de33d
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
vllm/config.py
vllm/config.py
+4
-2
No files found.
vllm/config.py
View file @
6dd5e528
...
...
@@ -986,6 +986,9 @@ class ModelConfig:
@
property
def
use_mla
(
self
)
->
bool
:
if
not
self
.
is_deepseek_mla
or
envs
.
VLLM_MLA_DISABLE
:
return
False
if
self
.
quantization
is
not
None
and
self
.
quantization
not
in
[
\
"fp8"
,
"compressed-tensors"
]:
logger
.
warning
(
...
...
@@ -1012,8 +1015,7 @@ class ModelConfig:
quant_config
)
return
False
use_mla
=
(
self
.
is_deepseek_mla
and
not
envs
.
VLLM_MLA_DISABLE
)
return
use_mla
return
True
@
property
def
supported_runner_types
(
self
)
->
Set
[
RunnerType
]:
...
...
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