From 46c63b8b0c05e144ed1f3edab92633821bad6361 Mon Sep 17 00:00:00 2001
From: Nathan Azrak <nathan.azrak@gmail.com>
Date: Fri, 20 Oct 2023 20:30:12 +1100
Subject: [PATCH] Adds mpnet to ORTConfigManager and NormalizedConfigManager

---
 optimum/onnxruntime/utils.py       | 1 +
 optimum/utils/normalized_config.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/optimum/onnxruntime/utils.py b/optimum/onnxruntime/utils.py
index cdd4460fb0f..ab129240205 100644
--- a/optimum/onnxruntime/utils.py
+++ b/optimum/onnxruntime/utils.py
@@ -123,6 +123,7 @@ class ORTConfigManager:
         "marian": "bart",
         "mbart": "bart",
         "mistral": "gpt2",
+        "mpnet": "bert",
         "mt5": "bart",
         "m2m-100": "bart",
         "nystromformer": "bert",
diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py
index 340fea4f1f9..73b39535943 100644
--- a/optimum/utils/normalized_config.py
+++ b/optimum/utils/normalized_config.py
@@ -234,6 +234,7 @@ class NormalizedConfigManager:
         "marian": BartLikeNormalizedTextConfig,
         "mbart": BartLikeNormalizedTextConfig,
         "mistral": MistralNormalizedTextConfig,
+        "mpnet": NormalizedTextConfig,
         "mt5": T5LikeNormalizedTextConfig,
         "m2m-100": BartLikeNormalizedTextConfig,
         "nystromformer": NormalizedTextConfig,