From e41bdd760b9aa5948c20fa69c5bbd9a2a03c69ae Mon Sep 17 00:00:00 2001 From: Qubitium-ModelCloud Date: Sat, 8 Feb 2025 00:14:54 +0800 Subject: [PATCH] make lint pass --- vllm/model_executor/layers/quantization/gptq_marlin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/quantization/gptq_marlin.py b/vllm/model_executor/layers/quantization/gptq_marlin.py index 6f1be14c9be60..d910587eb3b59 100644 --- a/vllm/model_executor/layers/quantization/gptq_marlin.py +++ b/vllm/model_executor/layers/quantization/gptq_marlin.py @@ -208,7 +208,7 @@ def get_quant_method( layer, ParallelLMHead) and self.lm_head_quantized if isinstance(layer, LinearBase) or parallel_lm_head_quantized: # False = skip module, None = no override, else = Positive match - if self.get_dynamic_override(layer_name=prefix) == False: + if self.get_dynamic_override(layer_name=prefix) == False: # noqa: E712 return UnquantizedEmbeddingMethod( ) if parallel_lm_head_quantized else UnquantizedLinearMethod()