Fix passing None as concrete args (#14022)

fde4867f · Thomas Wang · GitHub · 9eda0d15 · fde4867f
Unverified Commit fde4867f authored Oct 19, 2021 by Thomas Wang Committed by GitHub Oct 19, 2021
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/transformers/utils/fx.py src/transformers/utils/fx.py +3 -0

No files found.
--- a/src/transformers/utils/fx.py
+++ b/src/transformers/utils/fx.py
@@ -358,6 +358,9 @@ class HFTracer(Tracer):
        return attr_val

    def trace(self, root: PreTrainedModel, concrete_args: Optional[Dict[str, Any]] = None, method_names=None) -> Graph:
+        if concrete_args is None:
+            concrete_args = {}
+
        sig = inspect.signature(root.forward)
        input_names = sig.parameters.keys() - concrete_args.keys()