diff --git a/src/brevitas/proxy/runtime_quant.py b/src/brevitas/proxy/runtime_quant.py index 512dde45e..72124a6a4 100644 --- a/src/brevitas/proxy/runtime_quant.py +++ b/src/brevitas/proxy/runtime_quant.py @@ -197,7 +197,7 @@ def forward(self, x: Union[Tensor, QuantTensor]) -> Union[Tensor, QuantTensor]: class ActQuantProxyFromInjector(ActQuantProxyFromInjectorBase): def __init__(self, quant_layer, quant_injector): - super().__init__(self, quant_layer, quant_injector) + super().__init__(quant_layer, quant_injector) self.cache_class = _CachedIO def scale(self, force_eval=True):