From bc173b51b4a69c88a3d5d7726eba9db2223cb623 Mon Sep 17 00:00:00 2001 From: Lei Zhang Date: Wed, 4 Oct 2023 18:37:36 +0000 Subject: [PATCH] python lint --- .../tools/transformers/models/whisper/whisper_chain.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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,