Fixed inputs error for smoothquant example_inputs (#1473)

Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
intel · Dec 18, 2023 · 39f63a8 · 39f63a8
1 parent 33ece90
commit 39f63a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/neural_compressor/adaptor/torch_utils/smooth_quant.py b/neural_compressor/adaptor/torch_utils/smooth_quant.py
@@ -56,10 +56,11 @@ def move_input_to_device(input, device=torch.device("cpu")):
             tmp_input[k] = move_input_to_device(inp, device)
         input = tmp_input
     elif isinstance(input, list) or isinstance(input, tuple):
+        is_tuple = isinstance(input, tuple)
         tmp_input = []
         for inp in input:
             tmp_input.append(move_input_to_device(inp, device))
-        input = tmp_input
+        input = tuple(tmp_input) if is_tuple else tmp_input
     elif isinstance(input, torch.Tensor):
         input = input.to(device)  # pylint: disable=no-member
     return input