How to use jacquelinehe/mabel-roberta-base with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, RobertaForMabel tokenizer = AutoTokenizer.from_pretrained("jacquelinehe/mabel-roberta-base") model = RobertaForMabel.from_pretrained("jacquelinehe/mabel-roberta-base", device_map="auto")
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, RobertaForMabel tokenizer = AutoTokenizer.from_pretrained("jacquelinehe/mabel-roberta-base") model = RobertaForMabel.from_pretrained("jacquelinehe/mabel-roberta-base", device_map="auto")