vllm - 💡(How to fix) Fix [Bug]: KeyError: 'layers.0.mlp.gate_up_proj.g_idx' of GLM-OCR GPTQ Int8 in v0.21.1rc1

Official PRs (…)
ON THIS PAGE

Recommended Tools

×6

Utilities matched from this issue’s tags and category — try them while you read without losing context.

GitHub issue graph ai analysis

Paste a GitHub issue URL. We fetch that issue, discover linked issues from bodies/comments/timeline, collect linked pull requests, and produce a structured English report.

The report is written in English Markdown for sharing and archival.

Helpful · Quick feedback

Loading…

Error Message

(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] EngineCore failed to start. (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] Traceback (most recent call last): (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 1139, in run_engine_core (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] engine_core = EngineCoreProc(*args, engine_index=dp_rank, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return func(*args, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 905, in init (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] super().init( (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 121, in init (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] self.model_executor = executor_class(vllm_config) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return func(*args, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/abstract.py", line 109, in init (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] self._init_executor() (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/uniproc_executor.py", line 65, in _init_executor (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] self.driver_worker.load_model() (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 342, in load_model (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] self.model_runner.load_model(load_dummy_weights=load_dummy_weights) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return func(*args, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5053, in load_model (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] self.model = model_loader.load_model( (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return func(*args, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/base_loader.py", line 64, in load_model (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] self.load_weights(model, model_config) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return func(*args, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/default_loader.py", line 394, in load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] loaded_weights = model.load_weights(self.get_all_weights(model_config, model)) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4_1v.py", line 1696, in load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return original_load_weights(self, weights, *args, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] autoloaded_weights = set(self._load_module("", self.module, weights)) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] yield from self._load_module( (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] loaded_params = module_load_weights(weights) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 379, in load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return loader.load_weights(weights) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] return original_load_weights(self, weights, *args, **kwargs) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] autoloaded_weights = set(self._load_module("", self.module, weights)) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] yield from self._load_module( (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] loaded_params = module_load_weights(weights) (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 301, in load_weights (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] param = params_dict[name] (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] ~~~~~~~~~~~^^^^^^ (EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] KeyError: 'layers.0.mlp.gate_up_proj.g_idx'

Fix Action

Fix / Workaround

The error message

(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] EngineCore failed to start.
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] Traceback (most recent call last):
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 1139, in run_engine_core
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     engine_core = EngineCoreProc(*args, engine_index=dp_rank, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 905, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     super().__init__(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 121, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model_executor = executor_class(vllm_config)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                           ^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/abstract.py", line 109, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self._init_executor()
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/uniproc_executor.py", line 65, in _init_executor
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.driver_worker.load_model()
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 342, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model_runner.load_model(load_dummy_weights=load_dummy_weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5053, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model = model_loader.load_model(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                  ^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/base_loader.py", line 64, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.load_weights(model, model_config)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/default_loader.py", line 394, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_weights = model.load_weights(self.get_all_weights(model_config, model))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4_1v.py", line 1696, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return original_load_weights(self, weights, *args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     autoloaded_weights = set(self._load_module("", self.module, weights))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     yield from self._load_module(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_params = module_load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 379, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return loader.load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return original_load_weights(self, weights, *args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     autoloaded_weights = set(self._load_module("", self.module, weights))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     yield from self._load_module(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_params = module_load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 301, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     param = params_dict[name]
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]             ~~~~~~~~~~~^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] KeyError: 'layers.0.mlp.gate_up_proj.g_idx'

Possible cause of the bug

The weight file gate_up_proj was not split into gate_proj and up_proj. When the get_quant_method method of the GPTQMarlinConfig is executed during weight loading, the config.modules_in_block_to_quantize member is not split, causing it to return UnquantizedLinearMethod. UnquantizedLinearMethod only contains the weight parameter, so the g_idx key cannot be found when loading weights.

Code Example

Your output of `python collect_env.py` here

---

(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] EngineCore failed to start.
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] Traceback (most recent call last):
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 1139, in run_engine_core
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     engine_core = EngineCoreProc(*args, engine_index=dp_rank, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 905, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     super().__init__(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 121, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model_executor = executor_class(vllm_config)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                           ^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/abstract.py", line 109, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self._init_executor()
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/uniproc_executor.py", line 65, in _init_executor
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.driver_worker.load_model()
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 342, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model_runner.load_model(load_dummy_weights=load_dummy_weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5053, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model = model_loader.load_model(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                  ^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/base_loader.py", line 64, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.load_weights(model, model_config)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/default_loader.py", line 394, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_weights = model.load_weights(self.get_all_weights(model_config, model))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4_1v.py", line 1696, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return original_load_weights(self, weights, *args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     autoloaded_weights = set(self._load_module("", self.module, weights))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     yield from self._load_module(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_params = module_load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 379, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return loader.load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return original_load_weights(self, weights, *args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     autoloaded_weights = set(self._load_module("", self.module, weights))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     yield from self._load_module(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_params = module_load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 301, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     param = params_dict[name]
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]             ~~~~~~~~~~~^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] KeyError: 'layers.0.mlp.gate_up_proj.g_idx'

---

def configure_quant_config(
    quant_config: QuantizationConfig, model_class: type[nn.Module]
):
    """
    Pass packed_modules_mapping by reference to quant_config so that
    quant_config can properly match fused modules

    Note that model attributes are passed by reference to quant_config,
    enabling them to be updated by model_class.__new__ (ex. chatglm, qwen)

    Once the `SupportsQuant` mixin has been added to all models, this
    function can be removed
    """
    if not issubclass(model_class, SupportsQuant):
        hf_to_vllm_mapper = getattr(model_class, "hf_to_vllm_mapper", None)
        packed_mapping = getattr(model_class, "packed_modules_mapping", None)

        # pass mappings by reference to quant_config
        if hf_to_vllm_mapper is not None:
            quant_config.apply_vllm_mapper(hf_to_vllm_mapper)
        if packed_mapping is not None:
            quant_config.packed_modules_mapping = packed_mapping
            # fix suggestion
            replace_quantize_layers = []
            for layer in quant_config.modules_in_block_to_quantize:
                replace = False
                for key, val in packed_mapping.items():
                    if key in layer:
                        for v in val:
                            replace_quantize_layers.append(layer.replace(key, v))
                        replace = True
                if not replace:
                    replace_quantize_layers.append(layer)
            quant_config.modules_in_block_to_quantize = replace_quantize_layers
            # fix suggestion
RAW_BUFFERClick to expand / collapse

Your current environment

<details> <summary>The output of <code>python collect_env.py</code></summary> Name: vllm

Version: 0.21.1rc1.dev281+gaa2b56ffb

Summary: A high-throughput and memory-efficient inference and serving engine for LLMs

Home-page: https://github.com/vllm-project/vllm

Author: vLLM Team

Your output of `python collect_env.py` here
</details>

🐛 Describe the bug

Description

Failed to load glm-ocr gptq quantization model

vllm version: 0.21.1rc1.dev281+gaa2b56ffb

serve cmd

Run vllm serve ./GLM-OCR-w8g128 --served-model-name glm-ocr --tensor-parallel-size 1 --trust-remote-code --port 8888

Model information

https://huggingface.co/xiongjingwu/GLM-OCR-GPTQ-w8g128/tree/main

The error message

(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] EngineCore failed to start.
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] Traceback (most recent call last):
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 1139, in run_engine_core
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     engine_core = EngineCoreProc(*args, engine_index=dp_rank, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 905, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     super().__init__(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/engine/core.py", line 121, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model_executor = executor_class(vllm_config)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                           ^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/abstract.py", line 109, in __init__
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self._init_executor()
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/executor/uniproc_executor.py", line 65, in _init_executor
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.driver_worker.load_model()
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_worker.py", line 342, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model_runner.load_model(load_dummy_weights=load_dummy_weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/v1/worker/gpu_model_runner.py", line 5053, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.model = model_loader.load_model(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                  ^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/base_loader.py", line 64, in load_model
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     self.load_weights(model, model_config)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/tracing/otel.py", line 178, in sync_wrapper
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return func(*args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/default_loader.py", line 394, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_weights = model.load_weights(self.get_all_weights(model_config, model))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4_1v.py", line 1696, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return loader.load_weights(weights, mapper=self.hf_to_vllm_mapper)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return original_load_weights(self, weights, *args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     autoloaded_weights = set(self._load_module("", self.module, weights))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     yield from self._load_module(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_params = module_load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 379, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return loader.load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/model_loader/reload/torchao_decorator.py", line 50, in patched_model_load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     return original_load_weights(self, weights, *args, **kwargs)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 353, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     autoloaded_weights = set(self._load_module("", self.module, weights))
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 300, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     yield from self._load_module(
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/utils.py", line 273, in _load_module
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     loaded_params = module_load_weights(weights)
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]                     ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]   File "/usr/local/lib/python3.12/dist-packages/vllm/model_executor/models/glm4.py", line 301, in load_weights
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]     param = params_dict[name]
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165]             ~~~~~~~~~~~^^^^^^
(EngineCore pid=1202) ERROR 05-29 07:53:57 [core.py:1165] KeyError: 'layers.0.mlp.gate_up_proj.g_idx'

Possible cause of the bug

The weight file gate_up_proj was not split into gate_proj and up_proj. When the get_quant_method method of the GPTQMarlinConfig is executed during weight loading, the config.modules_in_block_to_quantize member is not split, causing it to return UnquantizedLinearMethod. UnquantizedLinearMethod only contains the weight parameter, so the g_idx key cannot be found when loading weights.

Therefore, this problem will occur if the gate_up_proj layer of a GPTQ quantized model is not split, such as the GLM-4.6V-Flash GPTQ quantization model.

Fix suggestion

In the configure_quant_config function of the vllm/model_executor/model_loader/utils.py file, if packed_mapping is not None, quant_config.modules_in_block_to_quantize should be updated.

def configure_quant_config(
    quant_config: QuantizationConfig, model_class: type[nn.Module]
):
    """
    Pass packed_modules_mapping by reference to quant_config so that
    quant_config can properly match fused modules

    Note that model attributes are passed by reference to quant_config,
    enabling them to be updated by model_class.__new__ (ex. chatglm, qwen)

    Once the `SupportsQuant` mixin has been added to all models, this
    function can be removed
    """
    if not issubclass(model_class, SupportsQuant):
        hf_to_vllm_mapper = getattr(model_class, "hf_to_vllm_mapper", None)
        packed_mapping = getattr(model_class, "packed_modules_mapping", None)

        # pass mappings by reference to quant_config
        if hf_to_vllm_mapper is not None:
            quant_config.apply_vllm_mapper(hf_to_vllm_mapper)
        if packed_mapping is not None:
            quant_config.packed_modules_mapping = packed_mapping
            # fix suggestion
            replace_quantize_layers = []
            for layer in quant_config.modules_in_block_to_quantize:
                replace = False
                for key, val in packed_mapping.items():
                    if key in layer:
                        for v in val:
                            replace_quantize_layers.append(layer.replace(key, v))
                        replace = True
                if not replace:
                    replace_quantize_layers.append(layer)
            quant_config.modules_in_block_to_quantize = replace_quantize_layers
            # fix suggestion

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

Vote matrix · Quick signals

Works
Did the solution work? Tap to confirm.
Easy Fix
Was it a quick fix?
Time Saver
Did it save you time?
Blocking
Was it severely blocking?
Common Issue
Are others likely hitting this too?
Flaky / Intermittent
Is it intermittent?
Verified / Reproducible
Can you reproduce it reliably?
Loading…

Still need to ship something?

×6

Another batch ranked right after the header list — different links, same matching logic.

Back to top recommendations

TRENDING

vllm - 💡(How to fix) Fix [Bug]: KeyError: 'layers.0.mlp.gate_up_proj.g_idx' of GLM-OCR GPTQ Int8 in v0.21.1rc1