From d2bf459f56dfbb351056e3021ec3400747333881 Mon Sep 17 00:00:00 2001
From: Maxim Bortnik <133897015+BortnikMaxim@users.noreply.github.com>
Date: Fri, 7 Aug 2026 19:43:39 +0300
Subject: [PATCH] Add LayoutLM to NormalizedConfigManager

---
 optimum/utils/normalized_config.py    |  2 +-
 tests/utils/test_normalized_config.py | 31 +++++++++++++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 tests/utils/test_normalized_config.py

diff --git a/optimum/utils/normalized_config.py b/optimum/utils/normalized_config.py
index 21470cb61f..5bbf5b5ab8 100644
--- a/optimum/utils/normalized_config.py
+++ b/optimum/utils/normalized_config.py
@@ -230,7 +230,6 @@ class NormalizedConfigManager:
         'groupvit',
         'hiera',
         'ibert',
-        'layoutlm',
         'layoutlmv3',
         'levit',
         'mobilebert',
@@ -276,6 +275,7 @@ class NormalizedConfigManager:
         "gptj": GPT2LikeNormalizedTextConfig,
         "imagegpt": GPT2LikeNormalizedTextConfig,
         "internlm2": NormalizedTextConfigWithGQA,
+        "layoutlm": NormalizedTextConfig,
         "llama": NormalizedTextConfigWithGQA,
         "longt5": T5LikeNormalizedTextConfig,
         "marian": BartLikeNormalizedTextConfig,
diff --git a/tests/utils/test_normalized_config.py b/tests/utils/test_normalized_config.py
new file mode 100644
index 0000000000..10b1e0dcfc
--- /dev/null
+++ b/tests/utils/test_normalized_config.py
@@ -0,0 +1,31 @@
+# coding=utf-8
+# Copyright 2022 The HuggingFace Team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+
+from transformers import LayoutLMConfig
+
+from optimum.utils.normalized_config import NormalizedConfigManager, NormalizedTextConfig
+
+
+class NormalizedConfigManagerTest(unittest.TestCase):
+    def test_layoutlm_is_registered(self):
+        config_class = NormalizedConfigManager.get_normalized_config_class("layoutlm")
+        self.assertIs(config_class, NormalizedTextConfig)
+
+        normalized_config = config_class(LayoutLMConfig(hidden_size=48, num_attention_heads=6, num_hidden_layers=3))
+        self.assertEqual(normalized_config.hidden_size, 48)
+        self.assertEqual(normalized_config.num_attention_heads, 6)
+        self.assertEqual(normalized_config.num_layers, 3)