Transformers How to use DylanJHJ/bert-base-final-v0-ep2 with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, BertForHighlightPrediction
tokenizer = AutoTokenizer.from_pretrained("DylanJHJ/bert-base-final-v0-ep2")
model = BertForHighlightPrediction.from_pretrained("DylanJHJ/bert-base-final-v0-ep2", device_map="auto") 
