[FX tracer] Make concrete_args from outside available (#21775)

make concrete_args from outside available
huggingface · Feb 27, 2023 · 2ea1ef9 · 2ea1ef9
1 parent ba2a5f1
commit 2ea1ef9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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()