KeyError: 'whisper'

#116
by aiyaqingzheng - opened

why???
Traceback (most recent call last):
File "ttt.py", line 12, in
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
File "/home/genliangzhao/.local/lib/python3.6/site-packages/transformers/models/auto/auto_factory.py", line 424, in from_pretrained
pretrained_model_name_or_path, return_unused_kwargs=True, trust_remote_code=trust_remote_code, **kwargs
File "/home/genliangzhao/.local/lib/python3.6/site-packages/transformers/models/auto/configuration_auto.py", line 672, in from_pretrained
config_class = CONFIG_MAPPING[config_dict["model_type"]]
File "/home/genliangzhao/.local/lib/python3.6/site-packages/transformers/models/auto/configuration_auto.py", line 387, in getitem
raise KeyError(key)
KeyError: 'whisper'

Hey @aiyaqingzheng - could you ensure that:

  1. You are using Python>=3.8 and torch>=1.11 (as per the docs)
  2. You are using the latest version of Transformers (pip install --upgrade transformers)

The traceback suggests you have an outdated version of Transformers!

Sign up or log in to comment