martinsinnona/plotqa
Viewer โข Updated โข 1.12k โข 183 โข 1
How to use anikde/semantic-label-aware-BERT_uncased with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, JointBERT
tokenizer = AutoTokenizer.from_pretrained("anikde/semantic-label-aware-BERT_uncased")
model = JointBERT.from_pretrained("anikde/semantic-label-aware-BERT_uncased", device_map="auto")