File tree 1 file changed +3
-1
lines changed
1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -791,9 +791,10 @@ def build_default_pretrains(default_treebanks):
791
791
# hfl bert: 0.9469
792
792
# hfl roberta: 0.9459
793
793
# hfl electra: 0.9515
794
+ # hfl macbert: 0.9530
794
795
# There is also a ShannonAI model, but our current codebase is
795
796
# somehow not compatible
796
- "zh-hans" : "hfl/chinese-electra-180g- large-discriminator " ,
797
+ "zh-hans" : "hfl/chinese-macbert- large" ,
797
798
}
798
799
799
800
TRANSFORMER_LAYERS = {
@@ -894,6 +895,7 @@ def build_default_pretrains(default_treebanks):
894
895
# zh
895
896
"google-bert/bert-base-chinese" : "google-bert-chinese" ,
896
897
"hfl/chinese-bert-wwm" : "hfl-bert-chinese" ,
898
+ "hfl/chinese-macbert-large" : "hfl-macbert-chinese" ,
897
899
"hfl/chinese-roberta-wwm-ext" : "hfl-roberta-chinese" ,
898
900
"hfl/chinese-electra-180g-large-discriminator" : "electra-large" ,
899
901
"ShannonAI/ChineseBERT-base" : "shannonai-chinese-bert" ,
You can’t perform that action at this time.
0 commit comments