Fix module path for symbolic_trace example

68a441fa · Sylvain Gugger · d3c3e722 · 68a441fa
Commit 68a441fa authored Jul 27, 2021 by Sylvain Gugger
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/utils/fx.py src/transformers/utils/fx.py +1 -1

No files found.
--- a/src/transformers/utils/fx.py
+++ b/src/transformers/utils/fx.py
@@ -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"],