diff --git a/onnxruntime/python/tools/transformers/models/whisper/whisper_chain.py b/onnxruntime/python/tools/transformers/models/whisper/whisper_chain.py index c0db8fb86c571..193039f71537a 100644 --- a/onnxruntime/python/tools/transformers/models/whisper/whisper_chain.py +++ b/onnxruntime/python/tools/transformers/models/whisper/whisper_chain.py @@ -1,13 +1,12 @@ import logging import os -import sys import onnx from onnx import TensorProto, helper from transformers import WhisperConfig -from benchmark_helper import Precision # noqa: E402 -from convert_generation import ( # noqa: E402 +from benchmark_helper import Precision +from convert_generation import ( get_shared_initializers, update_decoder_subgraph_output_cross_attention, update_decoder_subgraph_share_buffer_and_use_decoder_masked_mha,