nyu-mll/glue
Viewer • Updated • 1.49M • 429k • 522
How to use GeneZC/bert-base-stsb with Transformers:
# Load model directly
from transformers import AutoTokenizer, BertCls
tokenizer = AutoTokenizer.from_pretrained("GeneZC/bert-base-stsb")
model = BertCls.from_pretrained("GeneZC/bert-base-stsb", device_map="auto")