From b983ba35bd29f6d385efff8bedf80f7989c28d12 Mon Sep 17 00:00:00 2001 From: Enrique Shockwave <33002121+qeternity@users.noreply.github.com> Date: Thu, 14 Mar 2024 23:26:19 +0000 Subject: [PATCH] fix marlin config repr (#3414) --- vllm/model_executor/layers/quantization/marlin.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/quantization/marlin.py b/vllm/model_executor/layers/quantization/marlin.py index 0c4f20d9e3a5..48e44445a4a2 100644 --- a/vllm/model_executor/layers/quantization/marlin.py +++ b/vllm/model_executor/layers/quantization/marlin.py @@ -47,7 +47,7 @@ class MarlinConfig(QuantizationConfig): self.perm_len = 1024 def __repr__(self) -> str: - return f"MarlinConfig(group_size={self.group_size}" + return f"MarlinConfig(group_size={self.group_size})" @classmethod def get_name(cls) -> str: