couldn't load pretrained gpt2

[DeepInEvil]

It gives an error while loading gpt2
File "/home/debanjan/anaconda3/envs/py_3/lib/python3.7/site-packages/lm_scorer/models/gpt2.py", line 84, in _supported_model_names
return GPT2LMHeadModel.pretrained_model_archive_map.keys()
AttributeError: type object 'GPT2LMHeadModel' has no attribute 'pretrained_model_archive_map'

can you kindly check?

[simonepri]

Thanks for reporting, I'll have a look.

[DeepInEvil]

I did a quick fix, just import GPT2_PRETRAINED_CONFIG_ARCHIVE_MAP and return GPT2_PRETRAINED_CONFIG_ARCHIVE_MAP.keys() instead of GPT2LMHeadModel.pretrained_model_archive_map.keys()

[simonepri]

Fixed in v0.4.2.

Apparently they removed the pretrained_model_archive_map in huggingface/transformers#4152.

Thanks for reporting!