ArthurZ

/

jukebox-1b-lyrics

like 4

Hi,
It seems that the model cannot be loaded from AutoModel.from_pretrained. Are there any known workarounds to load the model? Thanks!

Code:

from transformers import AutoModel

model = AutoModel.from_pretrained("ArthurZ/jukebox-1b-lyrics")

Error:

/usr/local/lib/python3.7/dist-packages/transformers/models/auto/configuration_auto.py in __getitem__(self, key)
    430             return self._extra_content[key]
    431         if key not in self._mapping:
--> 432             raise KeyError(key)
    433         value = self._mapping[key]
    434         module_name = model_type_to_module_name(key)

KeyError: 'jukebox'