martinsinnona/plotqa
Viewer • Updated • 1.12k • 158 • 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")