changed modeling_fx_utils.py to utils/fx.py for clarity (#12326)

Co-authored-by: Michael Benayoun <michael@huggingface.co>
2021-06-23 18:16:24 +02:00
parent 941b4442ba
commit 986ac03e37
2 changed files with 3 additions and 3 deletions
--- a/tests/test_modeling_common.py
+++ b/tests/test_modeling_common.py
@@ -67,7 +67,7 @@ if is_torch_available():
    )

 if is_torch_fx_available():
-    from transformers.modeling_fx_utils import symbolic_trace
+    from transformers.utils.fx import symbolic_trace


 def _config_zero_init(config):