typo

pytorch · Apr 10, 2024 · 0bffa43 · 0bffa43
1 parent d5015b9
commit 0bffa43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quantize.py b/quantize.py
@@ -174,7 +174,7 @@ def dynamically_quantize_per_channel(
 
 
 
-def get_group_qparams(w, n_bit=4, groupsize=128, *, scales_dtype torch.float):
+def get_group_qparams(w, n_bit=4, groupsize=128, *, scales_dtype= torch.float):
     # needed for GPTQ with padding
     if groupsize > w.shape[-1]:
         groupsize = w.shape[-1]