typo

pytorch · Apr 16, 2024 · 9ba326e · 9ba326e
1 parent 7276777
commit 9ba326e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quantize.py b/quantize.py
@@ -743,7 +743,7 @@ def replace_linear_int4(
 class WeightOnlyInt4QuantHandler(QuantHandler):
     def __init__(self, mod, device, *, groupsize=128, inner_k_tiles=8, padding_allowed=True):
         self.mod = mod
-        self.device = device,
+        self.device = device
         self.groupsize = groupsize
         self.inner_k_tiles = inner_k_tiles
         self.padding_allowed = padding_allowed