diff --git a/src/brevitas/proxy/parameter_quant.py b/src/brevitas/proxy/parameter_quant.py index 204779505..516eea709 100644 --- a/src/brevitas/proxy/parameter_quant.py +++ b/src/brevitas/proxy/parameter_quant.py @@ -193,7 +193,7 @@ def scale(self): return out.scale def zero_point(self): - if self.requires_input_bit_width or not self.is_quant_enabled: + if not self.is_quant_enabled: return None zhs = self._zero_hw_sentinel() out = self.__call__(self.tracked_parameter_list[0], zhs, zhs)