update model card 7ebe97d
sebastian-hofstaetter commited on
How to use sebastian-hofstaetter/distilbert-cat-margin_mse-T2-msmarco with Transformers:
# Load model directly
from transformers import BERT_Cat
model = BERT_Cat.from_pretrained("sebastian-hofstaetter/distilbert-cat-margin_mse-T2-msmarco", device_map="auto")