How to use prajjwal1/bert-small with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("prajjwal1/bert-small", dtype="auto")
Hello!
From Transformers v5.2 onwards, it won't be possible to load this model without a model_type.
model_type
You can try this with
from transformers import AutoModel model = AutoModel.from_pretrained("prajjwal1/bert-small", revision="refs/pr/5") print(type(model))
· Sign up or log in to comment
