fix marlin config repr (#3414)

2024-03-14 23:26:19 +00:00
parent 54be8a0be2
commit b983ba35bd
1 changed files with 1 additions and 1 deletions
--- 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: