Transformers 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") 
