"Only symmetric quantization is supported for MoE"

#1
by ydawei - opened

Got this error, on vLLM 0.22.1:

EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/cohere2_moe.py", line 562, in __init__
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     self.model = Cohere2MoeModel(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]                  ^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/compilation/decorators.py", line 383, in __init__
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     old_init(self, *args, **kwargs)
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/cohere2_moe.py", line 402, in __init__
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     self.start_layer, self.end_layer, self.layers = make_layers(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]                                                     ^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 643, in make_layers
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     + get_offloader().wrap_modules(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/offloader/base.py", line 104, in wrap_modules
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     return list(modules_generator)
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 644, in <genexpr>
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     layer_fn(prefix=f"{prefix}.{idx}") for idx in range(start_layer, end_layer)
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/cohere2_moe.py", line 404, in <lambda>
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     lambda prefix: Cohere2MoeDecoderLayer(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]                    ^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/cohere2_moe.py", line 359, in __init__
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     self.mlp = Cohere2Moe(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]                ^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/cohere2_moe.py", line 300, in __init__
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     self.experts = FusedMoE(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]                    ^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/layers/fused_moe/layer.py", line 368, in __init__
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     self.quant_method: FusedMoEMethodBase = _get_quant_method()
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]                                             ^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/layers/fused_moe/layer.py", line 360, in _get_quant_method
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     quant_method = self.quant_config.get_quant_method(self, prefix)
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py", line 184, in get_quant_method
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     return CompressedTensorsMoEMethod.get_moe_method(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors_moe/compressed_tensors_moe.py", line 114, in get_moe_method
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     return CompressedTensorsWNA16MoEMethod(
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors_moe/compressed_tensors_moe_wna16.py", line 50, in __init__
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]     assert weight_quant.symmetric, (
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=83) ERROR 06-11 06:46:38 [core.py:1165] AssertionError: Only symmetric quantization is supported for MoE

Sign up or log in to comment