From ea6f9ce97e867f5772b876a3dbb2707b45cabb93 Mon Sep 17 00:00:00 2001 From: lygztq Date: Fri, 24 Feb 2023 06:09:25 +0000 Subject: [PATCH] make concrete_args from outside available --- src/transformers/utils/fx.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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()