Fixup mapping

This commit is contained in:
Julien Chaumond
2020-01-13 23:34:19 +00:00
parent b20f11d4ca
commit 894812c652

View File

@@ -60,8 +60,8 @@ TOKENIZER_MAPPING = OrderedDict(
(DistilBertConfig, DistilBertTokenizer), (DistilBertConfig, DistilBertTokenizer),
(AlbertConfig, AlbertTokenizer), (AlbertConfig, AlbertTokenizer),
(CamembertConfig, CamembertTokenizer), (CamembertConfig, CamembertTokenizer),
(RobertaConfig, XLMRobertaTokenizer), (RobertaConfig, RobertaTokenizer),
(XLMRobertaConfig, RobertaTokenizer), (XLMRobertaConfig, XLMRobertaTokenizer),
(BertConfig, BertTokenizer), (BertConfig, BertTokenizer),
(OpenAIGPTConfig, OpenAIGPTTokenizer), (OpenAIGPTConfig, OpenAIGPTTokenizer),
(GPT2Config, GPT2Tokenizer), (GPT2Config, GPT2Tokenizer),