remove wav2vec config

huggingface · May 21, 2024 · 55052eb · 55052eb
1 parent c3c9b3c
commit 55052eb
Showing 1 changed file with 0 additions and 16 deletions.
diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py
@@ -28,7 +28,6 @@
     UNetOnnxConfig,
     VaeDecoderOnnxConfig,
     VaeEncoderOnnxConfig,
-    Wav2Vec2OnnxConfig,
 )
 from optimum.exporters.tasks import TasksManager
 from optimum.utils import DEFAULT_DUMMY_SHAPES
@@ -91,21 +90,6 @@ def init_model_configs():
 register_in_tasks_manager = TasksManager.create_register("openvino", overwrite_existing=True)
 
 
-@register_in_tasks_manager(
-    "wav2vec2",
-    *[
-        "feature-extraction",
-        "automatic-speech-recognition",
-        "audio-classification",
-        "audio-frame-classification",
-        "audio-xvector",
-    ],
-    library_name="transformers",
-)
-class Wav2Vec2OpenVINOConfig(Wav2Vec2OnnxConfig):
-    DEFAULT_ONNX_OPSET = 14
-
-
 @register_in_tasks_manager("baichuan", *["text-generation", "text-generation-with-past"], library_name="transformers")
 class BaichaunOpenVINOConfig(TextDecoderWithPositionIdsOnnxConfig):
     DEFAULT_ONNX_OPSET = 13