fix vi-layoutxlm model name (#3129)

PaddlePaddle · Aug 24, 2022 · d3ace6c · d3ace6c
1 parent 949c848
commit d3ace6c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/paddlenlp/transformers/layoutlmv2/modeling.py b/paddlenlp/transformers/layoutlmv2/modeling.py
@@ -258,7 +258,7 @@ class LayoutLMv2PretrainedModel(PretrainedModel):
             "has_spatial_attention_bias": True,
             "has_visual_segment_embedding": False,
         },
-        "layoutlmv2-wo-backbone-base-uncased": {
+        "vi-layoutlmv2-base-uncased": {
             "attention_probs_dropout_prob": 0.1,
             "coordinate_size": 128,
             "fast_qkv": True,
@@ -297,8 +297,8 @@ class LayoutLMv2PretrainedModel(PretrainedModel):
             "https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/layoutlmv2-base-uncased/model_state.pdparams",
             "layoutlmv2-large-uncased":
             "https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/layoutlmv2-large-uncased/model_state.pdparams",
-            "layoutlmv2-wo-backbone-base-uncased":
-            "https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/layoutlmv2-wo-backbone-base-uncased/model_state.pdparams",
+            "vi-layoutlmv2-base-uncased":
+            "https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/vi-layoutlmv2-base-uncased/model_state.pdparams",
         }
     }
     base_model_prefix = "layoutlmv2"

diff --git a/paddlenlp/transformers/layoutxlm/modeling.py b/paddlenlp/transformers/layoutxlm/modeling.py
@@ -257,7 +257,7 @@ class LayoutXLMPretrainedModel(PretrainedModel):
             "type_vocab_size": 1,
             "vocab_size": 250002,
         },
-        "layoutxlm-wo-backbone-base-uncased": {
+        "vi-layoutxlm-base-uncased": {
             "attention_probs_dropout_prob": 0.1,
             "bos_token_id": 0,
             "coordinate_size": 128,
@@ -296,8 +296,8 @@ class LayoutXLMPretrainedModel(PretrainedModel):
         "model_state": {
             "layoutxlm-base-uncased":
             "https://bj.bcebos.com/paddlenlp/models/transformers/layoutxlm_base/model_state.pdparams",
-            "layoutxlm-wo-backbone-base-uncased":
-            "https://bj.bcebos.com/paddlenlp/models/transformers/layoutxlm_wo_backbone_base/model_state.pdparams",
+            "vi-layoutxlm-base-uncased":
+            "https://bj.bcebos.com/paddlenlp/models/transformers/vi-layoutxlm-base-uncased/model_state.pdparams",
         }
     }
     base_model_prefix = "layoutxlm"