Making auto-etc work better 7be6516
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")