Update README after RoBERTa addition

This commit is contained in:
Christophe Bourguignat authored and Lysandre Debut committed 2019-08-17 13:18:37 -04:00
1 parent d8923270e6
commit 189ff9b664
1 file changed
+1 -1
+1 -1
View File
@@ -76,7 +76,7 @@ import torch
from pytorch_transformers import *
# PyTorch-Transformers has a unified API
# for 6 transformer architectures and 27 pretrained weights.
# for 7 transformer architectures and 30 pretrained weights.
# Model | Tokenizer | Pretrained weights shortcut
MODELS = [(BertModel, BertTokenizer, 'bert-base-uncased'),
(OpenAIGPTModel, OpenAIGPTTokenizer, 'openai-gpt'),