We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents d83bf5b + de92ceb commit a579145Copy full SHA for a579145
mercy_transformer/config.py
@@ -2,4 +2,5 @@
2
3
MODELS = {
4
'bert': [BertModel, BertTokenizer, 'bert-base-multilingual-uncased'],
5
- 'distilbert': [DistilBertModel, DistilBertTokenizer, 'distilbert-base-multilingual-cased']}
+ 'distilbert': [DistilBertModel, DistilBertTokenizer, 'distilbert-base-multilingual-cased'],
6
+ 'albert': [AlbertModel, AlbertTokenizer, 'albert-base-v1']}
0 commit comments