Skip to content

GitLab

Unverified Commit 7963578f authored by Pavel Belevich's avatar Pavel Belevich Committed by GitHub
Browse files

Fix dummy_inputs() to dummy_inputs in symbolic_trace doc (#15776)

parent 074645e3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/utils/fx.py
View file @ 7963578f
...@@ -561,7 +561,7 @@ def symbolic_trace( ...@@ -561,7 +561,7 @@ def symbolic_trace(
model ([`PretrainedModel`]): model ([`PretrainedModel`]):
The model to trace. The model to trace.
input_names (`List[str]`, *optional*): input_names (`List[str]`, *optional*):
The names of the inputs of the traced model. If unset, model.dummy_inputs().keys() are used instead. The names of the inputs of the traced model. If unset, model.dummy_inputs.keys() are used instead.
Returns: Returns:
`torch.fx.GraphModule`: A GraphModule constructed by recording operations seen while tracing the model. `torch.fx.GraphModule`: A GraphModule constructed by recording operations seen while tracing the model.
......
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