From 79a843b413678d5ed14323846ae780cb84356e6d Mon Sep 17 00:00:00 2001
From: lijialin03 <lijialin03@baidu.com>
Date: Wed, 26 Nov 2025 06:17:13 +0000
Subject: [PATCH] apply MAPPING_SPACIAL_KEY to NAME_MAPPING

---
 paddleformers/transformers/auto/modeling.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/paddleformers/transformers/auto/modeling.py b/paddleformers/transformers/auto/modeling.py
index e60e287beb..6fe1a4452b 100644
--- a/paddleformers/transformers/auto/modeling.py
+++ b/paddleformers/transformers/auto/modeling.py
@@ -115,7 +115,10 @@ def get_name_mapping(task="Model"):
     """
     NAME_MAPPING = OrderedDict()
     for key, value in MAPPING_NAMES.items():
-        import_class = key + task
+        if key in MAPPING_SPACIAL_KEY and task == "Model":
+            import_class = MAPPING_SPACIAL_KEY[key] + task
+        else:
+            import_class = key + task
         new_key = key + "Model_Import_Class"
         NAME_MAPPING[new_key] = import_class
         NAME_MAPPING[import_class] = value