Fixed POT quantization

ekut-es · Nov 28, 2023 · 9dc985c · 9dc985c
1 parent af3ca10
commit 9dc985c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/hannah/quantization/qconfig.py b/hannah/quantization/qconfig.py
@@ -207,9 +207,7 @@ def __init__(
         self.dtype = dtype
 
         if power_of_2:
-            self.quantization_function = PowerOf2Quantization(
-                bits, scale=scale, debug=self.debug
-            )
+            self.quantization_function = PowerOf2Quantization(bits, debug=self.debug)
         else:
             self.quantization_function = SymmetricQuantization(
                 bits, rounding_mode=rounding_mode, debug=self.debug