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?

State Space Models org

It's not available yet! A PR is coming here: https://github.com/huggingface/transformers/pull/28094

Sign up or log in to comment