diff --git a/optimum/exporters/openvino/model_configs.py b/optimum/exporters/openvino/model_configs.py index 00269d1ba2..8feeafd619 100644 --- 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