Fix sglang bugs (#220)

Co-authored-by: gushiqiao <[email protected]>
ModelTC · Nov 22, 2024 · 7cd9b09 · 7cd9b09
1 parent f4fb949
commit 7cd9b09
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llmc/compression/quantization/module_utils.py b/llmc/compression/quantization/module_utils.py
@@ -1205,8 +1205,8 @@ def __repr__(self):
 
 
 class SglRealQuantLinear(VllmRealQuantLinear):
-    def __init__(self, weight, bias, scales, need_pack):
-        super().__init__(weight, bias, scales, need_pack)
+    def __init__(self, weight, bias, scales, input_scale, need_pack):
+        super().__init__(weight, bias, scales, input_scale, need_pack)
 
     def __repr__(self):
         return (