checkpoint15k 26853de
TalSchuster commited on
How to use tals/albert-base-vitaminc_rationale with Transformers:
# Load model directly
from transformers import AutoTokenizer, AlbertForTokenRationale
tokenizer = AutoTokenizer.from_pretrained("tals/albert-base-vitaminc_rationale")
model = AlbertForTokenRationale.from_pretrained("tals/albert-base-vitaminc_rationale", device_map="auto")