From 7a2e94a12036a8e8d9172fbf4638f6c976d13600 Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Wed, 18 Dec 2024 17:21:48 +0100
Subject: [PATCH] remove vit-mae masked-im task

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

diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
index bd8fd1de15..b09ec20a8c 100644
--- a/optimum/exporters/tasks.py
+++ b/optimum/exporters/tasks.py
@@ -1205,7 +1205,6 @@ class TasksManager:
         ),
         "vit-mae": supported_tasks_mapping(
             "feature-extraction",
-            "masked-im",
             onnx="VitMAEOnnxConfig",
         ),
         "vit-msn": supported_tasks_mapping(