Support automodel 307023a
Dan Fu commited on
How to use togethercomputer/m2-bert-80M-2k-retrieval with Transformers:
# Load model directly
from transformers import AutoModelForSequenceClassification
model = AutoModelForSequenceClassification.from_pretrained("togethercomputer/m2-bert-80M-2k-retrieval", trust_remote_code=True, device_map="auto")