diff --git a/sparsebit/quantization/quantizers/base.py b/sparsebit/quantization/quantizers/base.py index ec14d77..46d579b 100644 --- a/sparsebit/quantization/quantizers/base.py +++ b/sparsebit/quantization/quantizers/base.py @@ -89,7 +89,7 @@ def disable_quant(self): def enable_export_onnx(self): self.export_onnx = True # round zero point for onnx export - self.zero_point = self.zero_point.round() + self.zero_point.data = self.zero_point.data.round() def disable_export_onnx(self): self.export_onnx = False