SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Simpler Error message when loading config/model with .from_pretrained() (#3341)

Browse Source
This commit is contained in:
Julien Chaumond
2020-03-19 18:23:03 -04:00
committed by GitHub
parent 8eeefcb576
commit ecfd336318
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/configuration_utils.py
View File

@@ -251,14 +251,10 @@ class PretrainedConfig(object):
else: else:
msg = ( msg = (
"Can't load '{}'. Make sure that:\n\n" "Can't load '{}'. Make sure that:\n\n"
"- '{}' is either a correct model identifier of a community model from 'https://huggingface.co/models' which has a '{}' file\n\n" "- '{}' is a correct model identifier listed on 'https://huggingface.co/models'\n\n"
"- or '{}' is a model name in {}\n\n" "- or '{}' is the correct path to a directory containing a '{}' file\n\n".format(
"- or '{}' is the correct path to a directory containing a '{}' file".format(
pretrained_model_name_or_path, pretrained_model_name_or_path,
pretrained_model_name_or_path, pretrained_model_name_or_path,
CONFIG_NAME,
pretrained_model_name_or_path,
list(pretrained_config_archive_map.keys()),
pretrained_model_name_or_path, pretrained_model_name_or_path,
CONFIG_NAME, CONFIG_NAME,
) )