Initial release 7f8facc
Joeran Bosma commited on
How to use joeranbosma/dragon-roberta-base-domain-specific with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="joeranbosma/dragon-roberta-base-domain-specific") # Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("joeranbosma/dragon-roberta-base-domain-specific")
model = AutoModelForMaskedLM.from_pretrained("joeranbosma/dragon-roberta-base-domain-specific", device_map="auto")