Skip to content

Commit a579145

Browse files
authored
Merge pull request #5 from chagmgang/master
add albert to config
2 parents d83bf5b + de92ceb commit a579145

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mercy_transformer/config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,5 @@
22

33
MODELS = {
44
'bert': [BertModel, BertTokenizer, 'bert-base-multilingual-uncased'],
5-
'distilbert': [DistilBertModel, DistilBertTokenizer, 'distilbert-base-multilingual-cased']}
5+
'distilbert': [DistilBertModel, DistilBertTokenizer, 'distilbert-base-multilingual-cased'],
6+
'albert': [AlbertModel, AlbertTokenizer, 'albert-base-v1']}

0 commit comments

Comments
 (0)