fix issue

intel · Jun 4, 2024 · fe64795 · fe64795
1 parent db9ac04
commit fe64795
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/auto_round/quantizer.py b/auto_round/quantizer.py
@@ -85,8 +85,8 @@ def quant_weight_sym(weight, num_bits=4, v=0, min_scale=1.0, max_scale=1.0, scal
     """
     maxq = torch.tensor(2 ** num_bits - 1)
     if isinstance(min_scale, torch.Tensor):
-        wmin_tmp = torch.clamp(weight.min(1, keepdim=True)[0], max=0)
-        wmax_tmp = torch.clamp(weight.max(1, keepdim=True)[0], min=0)
+        wmin_tmp = torch.clamp(weight.min(1,)[0], max=0)
+        wmax_tmp = torch.clamp(weight.max(1)[0], min=0)
         wmin_tmp *= min_scale
         wmax_tmp *= max_scale
         wmax = torch.maximum(wmax_tmp, wmin_tmp)