[AutoTokenizer] Correct error message

This commit is contained in:
Julien Chaumond
2020-09-10 17:19:01 +02:00
parent db38f7ce29
commit d6c08b07a0

View File

@@ -234,7 +234,9 @@ class AutoTokenizer:
tokenizer_class_candidate = config.tokenizer_class
tokenizer_class = globals().get(tokenizer_class_candidate)
if tokenizer_class is None:
raise ValueError("Tokenizer class {} does not exist or is not currently imported.")
raise ValueError(
"Tokenizer class {} does not exist or is not currently imported.".format(tokenizer_class_candidate)
)
return tokenizer_class.from_pretrained(pretrained_model_name_or_path, *inputs, **kwargs)
# if model is an encoder decoder, the encoder tokenizer class is used by default