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
0ba1219a
Commit
0ba1219a
authored
Dec 24, 2025
by
yangql
Browse files
在v0.11中暂不支持awq的scale和zero合并操作
parent
fd8764b3
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
2 deletions
+4
-2
vllm/model_executor/layers/quantization/moe_wna16.py
vllm/model_executor/layers/quantization/moe_wna16.py
+2
-0
vllm/model_executor/model_loader/utils.py
vllm/model_executor/model_loader/utils.py
+2
-2
No files found.
vllm/model_executor/layers/quantization/moe_wna16.py
View file @
0ba1219a
...
@@ -195,6 +195,8 @@ class MoeWNA16Method(FusedMoEMethodBase):
...
@@ -195,6 +195,8 @@ class MoeWNA16Method(FusedMoEMethodBase):
moe
:
"FusedMoEConfig"
)
->
None
:
moe
:
"FusedMoEConfig"
)
->
None
:
super
().
__init__
(
moe
)
super
().
__init__
(
moe
)
self
.
quant_config
=
quant_config
self
.
quant_config
=
quant_config
# if os.getenv('AWQ_MOE_SZ') == None:
# os.environ['AWQ_MOE_SZ'] = '1'
self
.
use_w4a16_moe_sz
=
os
.
environ
.
get
(
'AWQ_MOE_SZ'
)
==
'1'
self
.
use_w4a16_moe_sz
=
os
.
environ
.
get
(
'AWQ_MOE_SZ'
)
==
'1'
self
.
use_w4a16_cuda
=
0
self
.
use_w4a16_cuda
=
0
self
.
use_moe_lmslim
=
0
self
.
use_moe_lmslim
=
0
...
...
vllm/model_executor/model_loader/utils.py
View file @
0ba1219a
...
@@ -244,8 +244,8 @@ def _get_model_architecture(
...
@@ -244,8 +244,8 @@ def _get_model_architecture(
# awq相关配置
# awq相关配置
try
:
try
:
if
os
.
getenv
(
'AWQ_MOE_SZ'
)
==
None
:
#
if os.getenv('AWQ_MOE_SZ') == None:
os
.
environ
[
'AWQ_MOE_SZ'
]
=
'1'
#
os.environ['AWQ_MOE_SZ'] = '1'
if
os
.
getenv
(
'AWQ_PAD'
)
==
None
and
(
torch
.
cuda
.
get_device_properties
(
torch
.
cuda
.
current_device
()).
multi_processor_count
==
120
):
if
os
.
getenv
(
'AWQ_PAD'
)
==
None
and
(
torch
.
cuda
.
get_device_properties
(
torch
.
cuda
.
current_device
()).
multi_processor_count
==
120
):
os
.
environ
[
'AWQ_PAD'
]
=
'1'
os
.
environ
[
'AWQ_PAD'
]
=
'1'
except
Exception
as
e
:
except
Exception
as
e
:
...
...
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