diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index 78976b9a042..e1162593082 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -1108,7 +1108,7 @@ class TasksManager: "text-to-audio", onnx="VitsOnnxConfig", ), - "visual_bert": supported_tasks_mapping( + "visual-bert": supported_tasks_mapping( "multiple-choice", "question-answering", "feature-extraction",