search

salesforce / CodeT5

Home of CodeT5: Open Code LLMs for Code Understanding and Generation
https://arxiv.org/abs/2305.07922
BSD 3-Clause "New" or "Revised" License
2.74k stars 401 forks source link

KeyError: 'codet5p' #113

Closed shailja-thakur closed 1 year ago

shailja-thakur commented 1 year ago

Getting the following error on the transformers-4.30.0.dev0

Traceback (most recent call last): File "gen_examples_HDLProblems_CodeT5.py", line 60, in tokenizer = AutoTokenizer.from_pretrained(checkpoint) File "/home/hp2265/.local/lib/python3.8/site-packages/transformers/models/auto/tokenization_auto.py", line 398, in from_pretrained config = AutoConfig.from_pretrained(pretrained_model_name_or_path, **kwargs) File "/home/hp2265/.local/lib/python3.8/site-packages/transformers/models/auto/configuration_auto.py", line 421, in from_pretrained config_class = CONFIG_MAPPING[config_dict["model_type"]] KeyError: 'codet5p'

yuewang-cuhk commented 1 year ago

Hi, can you provide a minimal example to reproduce the error? Besides, you can try to use the same version of transformers-4.21.3.