From 1f0933f472480659e2b1ece2908728da969a998d Mon Sep 17 00:00:00 2001
From: Nick Fraser <icanlosh@gmail.com>
Date: Wed, 4 Sep 2024 15:18:29 +0100
Subject: [PATCH] Fix (graph/gptq): Removed unnecessary QuantTensor check.

---
 src/brevitas/graph/gpxq.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/brevitas/graph/gpxq.py b/src/brevitas/graph/gpxq.py
index ce140842c..75992e8fe 100644
--- 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: