From 1185feb637b7c6ae9a02d8672c18d7b4e6eb0c15 Mon Sep 17 00:00:00 2001
From: Zane Neave <zaneneave@Zanes-MacBook-Pro.local>
Date: Tue, 11 Aug 2026 14:32:45 +1200
Subject: [PATCH] Add ConvBERT to NormalizedConfigManager. Register 'convbert'
 in NormalizedConfigManager._conf (maps to NormalizedTextConfig) and add it to
 TEXT_ENCODER_MODELS so the existing dummy-input-generator tests cover it.
 Addresses part of #351.

---
 optimum/utils/normalized_config.py          | 1 +
 tests/utils/test_dummpy_input_generators.py | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py
index 21470cb61f..7b15331db0 100644
--- a/optimum/utils/normalized_config.py
+++ b/optimum/utils/normalized_config.py
@@ -259,6 +259,7 @@ class NormalizedConfigManager:
         "falcon": NormalizedTextConfig,
         "camembert": NormalizedTextConfig,
         "codegen": GPT2LikeNormalizedTextConfig,
+        "convbert": NormalizedTextConfig,
         "cvt": NormalizedVisionConfig,
         "deberta": NormalizedTextConfig,
         "deberta-v2": NormalizedTextConfig,
diff --git a/tests/utils/test_dummpy_input_generators.py b/tests/utils/test_dummpy_input_generators.py
index e6fdf0b677..8601094ce6 100644
--- a/tests/utils/test_dummpy_input_generators.py
+++ b/tests/utils/test_dummpy_input_generators.py
@@ -30,7 +30,10 @@
     from optimum.utils.input_generators import DummyInputGenerator
 
 
-TEXT_ENCODER_MODELS = {"distilbert": "hf-internal-testing/tiny-random-DistilBertModel"}
+TEXT_ENCODER_MODELS = {
+    "convbert": "hf-internal-testing/tiny-random-ConvBertModel",
+    "distilbert": "hf-internal-testing/tiny-random-DistilBertModel",
+}
 
 VISION_MODELS = {"resnet": "hf-internal-testing/tiny-random-resnet"}