tner/bc5cdr
Viewer • Updated • 16.4k • 2.75k • 6
How to use sschet/bern2-ner with Transformers:
# Load model directly
from transformers import AutoTokenizer, RoBERTaMultiNER2
tokenizer = AutoTokenizer.from_pretrained("sschet/bern2-ner")
model = RoBERTaMultiNER2.from_pretrained("sschet/bern2-ner", device_map="auto")