From d7b5036f670cebb2e97986f440b4aba38cbd47b1 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Wed, 18 Dec 2024 21:35:33 +0000 Subject: [PATCH] fix for tensor --- src/brevitas/proxy/runtime_quant.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas/proxy/runtime_quant.py b/src/brevitas/proxy/runtime_quant.py index 32ecaa81e..6cd2c03ed 100644 --- a/src/brevitas/proxy/runtime_quant.py +++ b/src/brevitas/proxy/runtime_quant.py @@ -227,7 +227,7 @@ def create_quant_tensor( qt_args: Union[Tensor, Tuple[Any]], x: Optional[IntQuantTensor] = None) -> IntQuantTensor: - if x is None: + if isinstance(qt_args, tuple): out = IntQuantTensor(*qt_args, self.is_signed, self.training) else: out = IntQuantTensor(