diff --git a/torchao/quantization/pt2e/utils.py b/torchao/quantization/pt2e/utils.py index dc5f802fb8..114f6b0ab4 100644 --- a/torchao/quantization/pt2e/utils.py +++ b/torchao/quantization/pt2e/utils.py @@ -1031,6 +1031,8 @@ def replacement(x_i8, scale, zero_point, quant_min, quant_max): continue new_args = [] for arg in node.args: + if isinstance(arg, list): + arg = tuple(arg) # type: ignore[assignment] if ( _is_literal(arg) and arg not in exclude_literals