riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Fix module path for symbolic_trace example

Browse Source
This commit is contained in:
Sylvain Gugger 2021-07-27 13:47:22 -04:00
parent d3c3e722d6
commit 68a441fa4c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/utils/fx.py
View File

@ -357,7 +357,7 @@ def symbolic_trace(
Example::
from transformers.modeling_fx_utils import symbolic_trace
from transformers.utils.fx import symbolic_trace
traced_model = symbolic_trace(
model,
input_names=["input_ids", "attention_mask", "token_type_ids"],