From 335022d55b00e347139110a94dc16589dfc37ac3 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
Date: Tue, 3 Dec 2024 17:57:49 +0100
Subject: [PATCH] fix

---
 optimum/exporters/tasks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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",