eriktks/conll2003
Updated • 20.5k • 172
How to use tliu/asp-ner-flan-t5-base with Transformers:
# Load model directly
from transformers import T5NER
model = T5NER.from_pretrained("tliu/asp-ner-flan-t5-base", device_map="auto")This model is initialized from flan-t5-base and finetuned for named entity recognition task. The model structure is described in the paper Autoregressive Structured Prediction with Language Models, Github repo.
CUDA_VISIBLE_DEVICES=0 python evaluate_ner.py flant5_base tliu/asp-ner-flan-t5-base 0