From fb3407d013569f8ca4686e2946bc95a1fcfbaf1d Mon Sep 17 00:00:00 2001
From: IlyasMoutawwakil <moutawwakil.ilyas.tsi@gmail.com>
Date: Fri, 10 Jan 2025 16:26:43 +0100
Subject: [PATCH] disable modernbert

---
 optimum/exporters/tasks.py         | 18 +++++++++---------
 tests/exporters/exporters_utils.py |  4 ++--
 2 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py
index e8ea3830d3..6afe17a369 100644
--- a/optimum/exporters/tasks.py
+++ b/optimum/exporters/tasks.py
@@ -902,15 +902,15 @@ class TasksManager:
             "image-classification",
             onnx="MobileNetV2OnnxConfig",
         ),
-        "modernbert": supported_tasks_mapping(
-            "feature-extraction",
-            "fill-mask",
-            "text-classification",
-            "multiple-choice",
-            "token-classification",
-            "question-answering",
-            onnx="ModernBertOnnxConfig",
-        ),
+        # "modernbert": supported_tasks_mapping(
+        #     "feature-extraction",
+        #     "fill-mask",
+        #     "text-classification",
+        #     "multiple-choice",
+        #     "token-classification",
+        #     "question-answering",
+        #     onnx="ModernBertOnnxConfig",
+        # ),
         "mpnet": supported_tasks_mapping(
             "feature-extraction",
             "fill-mask",
diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py
index d256e16dd4..461e34adf6 100644
--- a/tests/exporters/exporters_utils.py
+++ b/tests/exporters/exporters_utils.py
@@ -125,7 +125,7 @@
     "mobilenet-v2": "hf-internal-testing/tiny-random-MobileNetV2Model",
     "mobilenet-v1": "google/mobilenet_v1_0.75_192",
     "mobilevit": "hf-internal-testing/tiny-random-mobilevit",
-    "modernbert": "hf-internal-testing/tiny-random-ModernBertForMaskedLM",
+    # "modernbert": "hf-internal-testing/tiny-random-ModernBertForMaskedLM",
     "mpnet": "hf-internal-testing/tiny-random-MPNetModel",
     "mpt": "hf-internal-testing/tiny-random-MptForCausalLM",
     "mt5": "lewtun/tiny-random-mt5",
@@ -267,7 +267,7 @@
     # "mobilenet_v1": "google/mobilenet_v1_0.75_192",
     # "mobilenet_v2": "google/mobilenet_v2_0.35_96",
     "mobilevit": "apple/mobilevit-small",
-    "modernbert": "answerdotai/ModernBERT-base",
+    # "modernbert": "answerdotai/ModernBERT-base",
     "mpt": "mosaicml/mpt-7b",
     "mt5": "lewtun/tiny-random-mt5",  # Not using google/mt5-small because it takes too much time for testing.
     "musicgen": "facebook/musicgen-small",