style

huggingface · Oct 6, 2023 · 05de3e3 · 05de3e3
1 parent 3dc1b85
commit 05de3e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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