diff --git a/optimum/onnxruntime/utils.py b/optimum/onnxruntime/utils.py index be395927cc..be7ec1f5d5 100644 --- a/optimum/onnxruntime/utils.py +++ b/optimum/onnxruntime/utils.py @@ -134,6 +134,7 @@ class ORTConfigManager: "marian": "bart", "mbart": "bart", "mistral": "gpt2", + "modernbert": "bert", "mpnet": "bert", "mt5": "bart", "m2m-100": "bart", diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py index 3f497b5920..5132143a23 100644 --- a/optimum/utils/normalized_config.py +++ b/optimum/utils/normalized_config.py @@ -264,6 +264,7 @@ class NormalizedConfigManager: "mbart": BartLikeNormalizedTextConfig, "mistral": NormalizedTextConfigWithGQA, "mixtral": NormalizedTextConfigWithGQA, + "modernbert": NormalizedTextConfig, "mpnet": NormalizedTextConfig, "mpt": MPTNormalizedTextConfig, "mt5": T5LikeNormalizedTextConfig,