Fix load config for UIE (#4271)

* fix load config

* update comment

Author: linjieccc

paddlenlp/taskflow/information_extraction.py

@@ -26,7 +26,7 @@
 from ..layers import GlobalPointerForEntityExtraction, GPLinkerForRelationExtraction
 from ..transformers import UIE, UIEM, UIEX, AutoModel, AutoTokenizer
 from ..utils.doc_parser import DocParser
-from ..utils.env import CONFIG_NAME
+from ..utils.env import CONFIG_NAME, LEGACY_CONFIG_NAME
 from ..utils.ie_utils import map_offset, pad_image_data
 from ..utils.log import logger
 from ..utils.tools import get_bool_ids_greater_than, get_span
@@ -114,7 +114,7 @@ class UIETask(Task):
 
     resource_files_names = {
         "model_state": "model_state.pdparams",
-        "model_config": "model_config.json",
+        "config": "config.json",
         "vocab_file": "vocab.txt",
         "special_tokens_map": "special_tokens_map.json",
         "tokenizer_config": "tokenizer_config.json",
@@ -126,9 +126,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base_v1.1/model_state.pdparams",
                 "47b93cf6a85688791699548210048085",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/model_config.json",
-                "a36c185bfc17a83b6cfef6f98b29c909",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/config.json",
+                "ad8b5442c758fb2dc18ea53b61e867f7",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/vocab.txt",
@@ -148,9 +148,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_medium_v1.1/model_state.pdparams",
                 "c34475665eb05e25f3c9cd9b020b331a",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_medium/model_config.json",
-                "6f1ee399398d4f218450fbbf5f212b15",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_medium/config.json",
+                "7fb22b3e07c5af76371c25ab814f06b8",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/vocab.txt",
@@ -170,9 +170,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_mini_v1.1/model_state.pdparams",
                 "9a0805762c41b104d590c15fbe9b19fd",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_mini/model_config.json",
-                "9229ce0a9d599de4602c97324747682f",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_mini/config.json",
+                "8ddebbf64c3f32a49e6f9e1c220e7322",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/vocab.txt",
@@ -192,9 +192,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_micro_v1.1/model_state.pdparams",
                 "da67287bca2906864929e16493f748e4",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_micro/model_config.json",
-                "07ef444420c3ab474f9270a1027f6da5",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_micro/config.json",
+                "544ddc65c758536cd3ba122f55b8709c",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/vocab.txt",
@@ -214,9 +214,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_nano_v1.1/model_state.pdparams",
                 "48db5206232e89ef16b66467562d90e5",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_nano/model_config.json",
-                "e3a9842edf8329ccdd0cf6039cf0a8f8",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_nano/config.json",
+                "e0e0a2c0d9651ed1a8492be5507590a9",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/vocab.txt",
@@ -237,9 +237,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_medium_v1.1/model_state.pdparams",
                 "c34475665eb05e25f3c9cd9b020b331a",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_medium/model_config.json",
-                "6f1ee399398d4f218450fbbf5f212b15",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_medium/config.json",
+                "7fb22b3e07c5af76371c25ab814f06b8",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/vocab.txt",
@@ -259,9 +259,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_medical_base_v0.2/model_state.pdparams",
                 "7582d3b01f6faf00b7000111ea853796",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/model_config.json",
-                "a36c185bfc17a83b6cfef6f98b29c909",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/config.json",
+                "ad8b5442c758fb2dc18ea53b61e867f7",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base/vocab.txt",
@@ -281,9 +281,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base_en_v1.2/model_state.pdparams",
                 "8c5d5c8faa76681a0aad58f982cd6141",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base_en/model_config.json",
-                "2ca9fe0eea8ff9418725d1a24fcf5c36",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base_en/config.json",
+                "257b80ea8b7889fd8b83a9ace7a8a220",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_base_en/vocab.txt",
@@ -303,9 +303,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_base_v1.1/model_state.pdparams",
                 "eb00c06bd7144e76343d750f5bf36ff6",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_base/model_config.json",
-                "05c4b9d050e1402a891b207e36d2e501",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_base/config.json",
+                "f03de3ce1b83c13e7bee18e6f323d33f",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_base/vocab.txt",
@@ -329,9 +329,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_large_v1.1/model_state.pdparams",
                 "9db83a67f34a9c2483dbe57d2510b4c2",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_large/model_config.json",
-                "22ad69618dc3f4c3fe756e3044c3056e",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_large/config.json",
+                "8f540de05de57ecc66336b41f3a7ffdb",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_m_large/vocab.txt",
@@ -355,9 +355,9 @@ class UIETask(Task):
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_x_base_v1.0/model_state.pdparams",
                 "a953b55f7639ae73d1df6c2c5f7667dd",
             ],
-            "model_config": [
-                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_x_base/model_config.json",
-                "50be05e78aec34d37596513870fa050e",
+            "config": [
+                "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_x_base/config.json",
+                "6bcd7d4b119717121fa0276c20bd9224",
             ],
             "vocab_file": [
                 "https://bj.bcebos.com/paddlenlp/taskflow/information_extraction/uie_x_base/vocab.txt",
@@ -403,9 +403,17 @@ def __init__(self, task, model, schema=None, **kwargs):
             with open(config_file_path) as f:
                 self._init_class = json.load(f)["architectures"].pop()
         else:
-            self._check_task_files()
-            with open(os.path.join(self._task_path, "model_config.json")) as f:
-                self._init_class = json.load(f)["init_class"]
+            # Compatible with the model fine-tuned without PretrainedConfig
+            if os.path.exists(os.path.join(self._task_path, LEGACY_CONFIG_NAME)):
+                if "config" in self.resource_files_names.keys():
+                    del self.resource_files_names["config"]
+                with open(os.path.join(self._task_path, LEGACY_CONFIG_NAME)) as f:
+                    self._init_class = json.load(f)["init_class"]
+                self._check_task_files()
+            else:
+                self._check_task_files()
+                with open(os.path.join(self._task_path, CONFIG_NAME)) as f:
+                    self._init_class = json.load(f)["architectures"].pop()
 
         if self._init_class not in ["UIEX", "UIEM"]:
             if "sentencepiece_model_file" in self.resource_files_names.keys():