c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added RoBERTa tokenizer to AutoTokenizer

Browse Source
This commit is contained in:
LysandreJik committed 2019-08-15 17:07:07 -04:00
1 parent e24e19ce3b
commit 83dba0b67b
2 files changed
+8 -3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

pytorch_transformers/modeling_auto.py
+2 -2
View File
@@ -127,7 +127,7 @@ class AutoConfig(object):
raise ValueError("Unrecognized model identifier in {}. Should contains one of "
"'bert', 'openai-gpt', 'gpt2', 'transfo-xl', 'xlnet', "
"'xlm'".format(pretrained_model_name_or_path))
"'xlm', 'roberta'".format(pretrained_model_name_or_path))
class AutoModel(object):
@@ -242,4 +242,4 @@ class AutoModel(object):
raise ValueError("Unrecognized model identifier in {}. Should contains one of "
"'bert', 'openai-gpt', 'gpt2', 'transfo-xl', 'xlnet', "
"'xlm'".format(pretrained_model_name_or_path))
"'xlm', 'roberta'".format(pretrained_model_name_or_path))