Fix for passthrough

Xilinx · Jan 8, 2024 · d606de9 · d606de9
1 parent 2f7f216
commit d606de9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/proxy/runtime_quant.py b/src/brevitas/proxy/runtime_quant.py
@@ -158,7 +158,7 @@ def forward(self, x: Union[Tensor, QuantTensor]) -> QuantTensor:
                 if isinstance(y, tuple):
                     y = y[0]
                 if isinstance(x, QuantTensor):
-                    return QuantTensor(
+                    return QuantTensor.from_fake_quantized(
                         y, x.scale, x.zero_point, x.bit_width, x.signed, self.training)
                 else:
                     return y