Skip to content

Commit 55052eb

Browse files
committed
remove wav2vec config
1 parent c3c9b3c commit 55052eb

File tree

1 file changed

+0
-16
lines changed

1 file changed

+0
-16
lines changed

optimum/exporters/openvino/model_configs.py

-16
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
UNetOnnxConfig,
2929
VaeDecoderOnnxConfig,
3030
VaeEncoderOnnxConfig,
31-
Wav2Vec2OnnxConfig,
3231
)
3332
from optimum.exporters.tasks import TasksManager
3433
from optimum.utils import DEFAULT_DUMMY_SHAPES
@@ -91,21 +90,6 @@ def init_model_configs():
9190
register_in_tasks_manager = TasksManager.create_register("openvino", overwrite_existing=True)
9291

9392

94-
@register_in_tasks_manager(
95-
"wav2vec2",
96-
*[
97-
"feature-extraction",
98-
"automatic-speech-recognition",
99-
"audio-classification",
100-
"audio-frame-classification",
101-
"audio-xvector",
102-
],
103-
library_name="transformers",
104-
)
105-
class Wav2Vec2OpenVINOConfig(Wav2Vec2OnnxConfig):
106-
DEFAULT_ONNX_OPSET = 14
107-
108-
10993
@register_in_tasks_manager("baichuan", *["text-generation", "text-generation-with-past"], library_name="transformers")
11094
class BaichaunOpenVINOConfig(TextDecoderWithPositionIdsOnnxConfig):
11195
DEFAULT_ONNX_OPSET = 13

0 commit comments

Comments
 (0)