fix smoothquant issue on fetching example_inputs (#1129)

Signed-off-by: Xin He <[email protected]>
intel · Aug 4, 2023 · df8d6ec · df8d6ec
1 parent c93e16b
commit df8d6ec
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/neural_compressor/adaptor/torch_utils/smooth_quant.py b/neural_compressor/adaptor/torch_utils/smooth_quant.py
@@ -832,6 +832,7 @@ def _get_example_input(self):
             ##assert self.dataloader, "Please provide dataloader or example_inputs"
             for idx, input in enumerate(self.dataloader):
                 self.example_inputs = input
+                break
 
         return self.example_inputs