Fix (graph/gptq): Removed unnecessary QuantTensor check.

Xilinx · Sep 4, 2024 · 1f0933f · 1f0933f
1 parent 4b1377a
commit 1f0933f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/brevitas/graph/gpxq.py b/src/brevitas/graph/gpxq.py
@@ -238,8 +238,7 @@ def process_input(self, inp):
         if isinstance(inp, IntQuantTensor):
             if is_quant_enabled and self.quant_metadata is None:
                 self.quant_metadata = _CachedIO(inp, metadata_only=True)
-            if isinstance(inp, QuantTensor):
-                inp = inp.value
+            inp = inp.value
 
         # If input is unbatched, add batch_size = 1
         if len(inp.shape) == 1: