Fix barthez tokenizer (#9562)

This commit is contained in:
Lysandre Debut
2021-01-13 06:24:10 -05:00
committed by GitHub
parent 247a7b2029
commit 245cdb469d

View File

@@ -199,7 +199,6 @@ TOKENIZER_MAPPING = OrderedDict(
(AlbertConfig, (AlbertTokenizer, AlbertTokenizerFast)),
(CamembertConfig, (CamembertTokenizer, CamembertTokenizerFast)),
(PegasusConfig, (PegasusTokenizer, PegasusTokenizerFast)),
(MBartConfig, (BarthezTokenizer, BarthezTokenizerFast)),
(MBartConfig, (MBartTokenizer, MBartTokenizerFast)),
(XLMRobertaConfig, (XLMRobertaTokenizer, XLMRobertaTokenizerFast)),
(MarianConfig, (MarianTokenizer, None)),
@@ -243,6 +242,7 @@ NO_CONFIG_TOKENIZER = [
HerbertTokenizer,
HerbertTokenizerFast,
PhobertTokenizer,
BarthezTokenizer,
]