Make convert_to_onnx runable as script again (huggingface#20009)

* Make convert_to_onnx runable as script again Fix `convert_graph_to_onnx.py` relative import so it can be run as a script again. * Trigger CI
amyeroberts · Dec 7, 2022 · dcf3c21 · dcf3c21
1 parent a904c13
commit dcf3c21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/convert_graph_to_onnx.py b/src/transformers/convert_graph_to_onnx.py
@@ -273,7 +273,7 @@ def convert_pytorch(nlp: Pipeline, opset: int, output: Path, use_external_format
     import torch
     from torch.onnx import export
 
-    from .pytorch_utils import is_torch_less_than_1_11
+    from transformers.pytorch_utils import is_torch_less_than_1_11
 
     print(f"Using framework PyTorch: {torch.__version__}")