Skip to content

Commit d911bf2

Browse files
authored
Change to pretrained weight to ernie-1.0-large-zh-cw (#2400)
* change to ernie-1.0-large-pp-zh * change to ernie-1.0-large-zh-cw
1 parent f428a8c commit d911bf2

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

paddlenlp/transformers/ernie/modeling.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ class ErniePretrainedModel(PretrainedModel):
146146
"vocab_size": 18000,
147147
"pad_token_id": 0,
148148
},
149-
"ernie-1.0-large-zh": {
149+
"ernie-1.0-large-zh-cw": {
150150
"attention_probs_dropout_prob": 0.1,
151151
"hidden_act": "relu",
152152
"hidden_dropout_prob": 0.1,
@@ -415,8 +415,8 @@ class ErniePretrainedModel(PretrainedModel):
415415
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie/ernie_v1_chn_base.pdparams",
416416
"ernie-1.0-base-zh":
417417
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie/ernie_v1_chn_base.pdparams",
418-
"ernie-1.0-large-zh":
419-
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie/ernie_1.0_large_zh.pdparams",
418+
"ernie-1.0-large-zh-cw":
419+
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie/ernie_1.0_large_zh_cw.pdparams",
420420
"ernie-tiny":
421421
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie_tiny/ernie_tiny.pdparams",
422422
"ernie-2.0-base-zh":

paddlenlp/transformers/ernie/tokenizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ class ErnieTokenizer(PretrainedTokenizer):
8383
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie/vocab.txt",
8484
"ernie-1.0-base-zh":
8585
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie/vocab.txt",
86-
"ernie-1.0-large-zh":
86+
"ernie-1.0-large-zh-cw":
8787
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie/vocab.txt",
8888
"ernie-tiny":
8989
"https://bj.bcebos.com/paddlenlp/models/transformers/ernie_tiny/vocab.txt",
@@ -136,7 +136,7 @@ class ErnieTokenizer(PretrainedTokenizer):
136136
"ernie-1.0-base-zh": {
137137
"do_lower_case": True
138138
},
139-
"ernie-1.0-large-zh": {
139+
"ernie-1.0-large-zh-cw": {
140140
"do_lower_case": True
141141
},
142142
"ernie-tiny": {

0 commit comments

Comments
 (0)