サブロウ丸

Sabrou-mal サブロウ丸

主にプログラミングと数学

KeyError for from_pretrained in transformers

I encountered the following error while trying to execute the microsoft/phi-1 model in HuggingFace transformers module.

Traceback (most recent call last):                                                                                                                            
  File "xxx.py", line 293, in <module>
    main(args)
  File "xxx.py", line 104, in main
    config = AutoConfig.from_pretrained(args.base_modelname,
  File "/usr/lib/python3/dist-packages/transformers/models/auto/configuration_auto.py", line 1050, in from_pretrained
    config_class = CONFIG_MAPPING[config_dict["model_type"]]
  File "/usr/lib/python3/dist-packages/transformers/models/auto/configuration_auto.py", line 748, in __getitem__
    raise KeyError(key)
KeyError: 'phi'

I resolved this error by upgrading the transformers version....

pip install transformers --upgrade