Unverified Commit 8ea6694d authored by Mihai Cernusca's avatar Mihai Cernusca Committed by GitHub
Browse files

Make convert_to_onnx runable as script again (#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
parent 84c9bf74
......@@ -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__}")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment