From b880c8d5263b07e3ba37a8213e52cf79c1ffda9e Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Fri, 9 Feb 2024 16:14:21 +0000 Subject: [PATCH] fix --- src/brevitas/nn/mixin/parameter.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/brevitas/nn/mixin/parameter.py b/src/brevitas/nn/mixin/parameter.py index e3d6c421c..4e5a733f5 100644 --- a/src/brevitas/nn/mixin/parameter.py +++ b/src/brevitas/nn/mixin/parameter.py @@ -176,8 +176,7 @@ def quant_bias(self): if self.bias is None: return None scale = self.quant_bias_scale() - bit_width = self.quant_bias_bit_width() - quant_bias = self.bias_quant(self.bias, scale, bit_width) + quant_bias = self.bias_quant(self.bias, scale) return quant_bias def quant_bias_scale(self):