diff --git a/src/transformers/utils/fx.py b/src/transformers/utils/fx.py index 339ec1fc7831..da9c43b17195 100755 --- a/src/transformers/utils/fx.py +++ b/src/transformers/utils/fx.py @@ -984,7 +984,13 @@ def trace( continue if param.default is inspect.Parameter.empty: raise ValueError(f"You need to specify a default value for the parameter {param.name}.") - concrete_args.update({p.name: p.default for p in sig.parameters.values() if p.name not in dummy_inputs}) + concrete_args.update( + { + p.name: p.default + for p in sig.parameters.values() + if (p.name not in dummy_inputs and p.name not in concrete_args) + } + ) input_names = sig.parameters.keys() - concrete_args.keys()