Error importing the pre-trained model
#2
by
poomchantara
- opened
I try to import
from transformers import AutoModel
model = AutoModel.from_pretrained("state-spaces/mamba-370m")
but get -> ValueError: Unrecognized model in state-spaces/mamba-370m. Should have a model_type
key in its config.json .....
How can I fix this?
It's not available yet! A PR is coming here: https://github.com/huggingface/transformers/pull/28094