From 05de3e37a994922aa377688bf36a6b464074f45f Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Fri, 6 Oct 2023 13:32:32 +0200 Subject: [PATCH] style --- optimum/pipelines/pipelines_base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/optimum/pipelines/pipelines_base.py b/optimum/pipelines/pipelines_base.py index 557bc089372..37430205855 100644 --- a/optimum/pipelines/pipelines_base.py +++ b/optimum/pipelines/pipelines_base.py @@ -19,7 +19,9 @@ from transformers import ( AudioClassificationPipeline, AutoConfig, + AutoFeatureExtractor, AutomaticSpeechRecognitionPipeline, + AutoTokenizer, FeatureExtractionPipeline, FillMaskPipeline, ImageClassificationPipeline, @@ -38,10 +40,8 @@ TranslationPipeline, ZeroShotClassificationPipeline, ) -from transformers import AutoTokenizer, AutoFeatureExtractor from transformers import pipeline as transformers_pipeline from transformers.feature_extraction_utils import PreTrainedFeatureExtractor -from transformers.onnx.utils import get_preprocessor from transformers.pipelines import SUPPORTED_TASKS as TRANSFORMERS_SUPPORTED_TASKS from transformers.pipelines import infer_framework_load_model