Skip to content

Commit d3ace6c

Browse files
fix vi-layoutxlm model name (#3129)
1 parent 949c848 commit d3ace6c

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

paddlenlp/transformers/layoutlmv2/modeling.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,7 @@ class LayoutLMv2PretrainedModel(PretrainedModel):
258258
"has_spatial_attention_bias": True,
259259
"has_visual_segment_embedding": False,
260260
},
261-
"layoutlmv2-wo-backbone-base-uncased": {
261+
"vi-layoutlmv2-base-uncased": {
262262
"attention_probs_dropout_prob": 0.1,
263263
"coordinate_size": 128,
264264
"fast_qkv": True,
@@ -297,8 +297,8 @@ class LayoutLMv2PretrainedModel(PretrainedModel):
297297
"https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/layoutlmv2-base-uncased/model_state.pdparams",
298298
"layoutlmv2-large-uncased":
299299
"https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/layoutlmv2-large-uncased/model_state.pdparams",
300-
"layoutlmv2-wo-backbone-base-uncased":
301-
"https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/layoutlmv2-wo-backbone-base-uncased/model_state.pdparams",
300+
"vi-layoutlmv2-base-uncased":
301+
"https://bj.bcebos.com/paddlenlp/models/transformers/layoutlmv2/vi-layoutlmv2-base-uncased/model_state.pdparams",
302302
}
303303
}
304304
base_model_prefix = "layoutlmv2"

paddlenlp/transformers/layoutxlm/modeling.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ class LayoutXLMPretrainedModel(PretrainedModel):
257257
"type_vocab_size": 1,
258258
"vocab_size": 250002,
259259
},
260-
"layoutxlm-wo-backbone-base-uncased": {
260+
"vi-layoutxlm-base-uncased": {
261261
"attention_probs_dropout_prob": 0.1,
262262
"bos_token_id": 0,
263263
"coordinate_size": 128,
@@ -296,8 +296,8 @@ class LayoutXLMPretrainedModel(PretrainedModel):
296296
"model_state": {
297297
"layoutxlm-base-uncased":
298298
"https://bj.bcebos.com/paddlenlp/models/transformers/layoutxlm_base/model_state.pdparams",
299-
"layoutxlm-wo-backbone-base-uncased":
300-
"https://bj.bcebos.com/paddlenlp/models/transformers/layoutxlm_wo_backbone_base/model_state.pdparams",
299+
"vi-layoutxlm-base-uncased":
300+
"https://bj.bcebos.com/paddlenlp/models/transformers/vi-layoutxlm-base-uncased/model_state.pdparams",
301301
}
302302
}
303303
base_model_prefix = "layoutxlm"

0 commit comments

Comments
 (0)