diff --git a/optimum/onnxruntime/modeling_decoder.py b/optimum/onnxruntime/modeling_decoder.py index 8e484b5d8ea..3b92c5dd341 100644 --- a/optimum/onnxruntime/modeling_decoder.py +++ b/optimum/onnxruntime/modeling_decoder.py @@ -20,6 +20,8 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type, Union import numpy as np +import onnx +import onnxruntime import torch from huggingface_hub import hf_hub_download from huggingface_hub.utils import EntryNotFoundError @@ -28,9 +30,6 @@ from transformers.file_utils import add_end_docstrings, add_start_docstrings_to_model_forward from transformers.modeling_outputs import CausalLMOutputWithPast -import onnx -import onnxruntime - from ..exporters import TasksManager from ..exporters.onnx import export, main_export from ..onnx.utils import _get_external_data_paths