perceptiveshawty/datasets-for-compcse
Preview • Updated • 7
How to use perceptiveshawty/compositional-bert-base-uncased with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, BertForCL
tokenizer = AutoTokenizer.from_pretrained("perceptiveshawty/compositional-bert-base-uncased")
model = BertForCL.from_pretrained("perceptiveshawty/compositional-bert-base-uncased", device_map="auto")