riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 1 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:: Example::
from transformers.modeling_fx_utils import symbolic_trace from transformers.utils.fx import symbolic_trace
traced_model = symbolic_trace( traced_model = symbolic_trace(
model, model,
input_names=["input_ids", "attention_mask", "token_type_ids"], input_names=["input_ids", "attention_mask", "token_type_ids"],