Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "e9c23fa056f401a586a1691edf773d1b9b60f96d"
Unverified Commit fde4867f authored by Thomas Wang's avatar Thomas Wang Committed by GitHub
Browse files

Fix passing None as concrete args (#14022)

parent 9eda0d15
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/transformers/utils/fx.py
View file @ fde4867f
...@@ -358,6 +358,9 @@ class HFTracer(Tracer): ...@@ -358,6 +358,9 @@ class HFTracer(Tracer):
return attr_val return attr_val
def trace(self, root: PreTrainedModel, concrete_args: Optional[Dict[str, Any]] = None, method_names=None) -> Graph: 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) sig = inspect.signature(root.forward)
input_names = sig.parameters.keys() - concrete_args.keys() input_names = sig.parameters.keys() - concrete_args.keys()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment