fix marlin config repr (#3414)

b983ba35 · Enrique Shockwave · GitHub · 54be8a0b · b983ba35
Unverified Commit b983ba35 authored Mar 14, 2024 by Enrique Shockwave Committed by GitHub Mar 14, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

vllm/model_executor/layers/quantization/marlin.py vllm/model_executor/layers/quantization/marlin.py +1 -1

No files found.
--- 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: