Skip to content

GitLab

Unverified Commit b5b957a6 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Make convert_to_onnx runable as script again (#10857)

parent 77bf3fe7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/transformers/convert_graph_to_onnx.py
View file @ b5b957a6
...@@ -19,9 +19,9 @@ from typing import Dict, List, Optional, Tuple ...@@ -19,9 +19,9 @@ from typing import Dict, List, Optional, Tuple
from packaging.version import Version, parse from packaging.version import Version, parse
from .file_utils import ModelOutput, is_tf_available, is_torch_available from transformers.file_utils import ModelOutput, is_tf_available, is_torch_available
from .pipelines import Pipeline, pipeline from transformers.pipelines import Pipeline, pipeline
from .tokenization_utils import BatchEncoding from transformers.tokenization_utils import BatchEncoding
# This is the minimal required version to # This is the minimal required version to
......
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