Add missing "model_type": "bert" to allow loading with Transformers v5.2

#2
by tomaarsen HF Staff - opened

Hello!

Details

From Transformers v5.2 onwards, it won't be possible to load this model without a model_type.

You can try this with

from transformers import AutoModel

model = AutoModel.from_pretrained("prajjwal1/bert-medium", revision="refs/pr/2")
print(type(model))
  • Tom Aarsen
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment