ontonotes/conll2012_ontonotesv5
Updated • 1.27k • 45
How to use tliu/asp-coref-t0-3b with Transformers:
# Load model directly
from transformers import AutoTokenizer, T5Coref
tokenizer = AutoTokenizer.from_pretrained("tliu/asp-coref-t0-3b")
model = T5Coref.from_pretrained("tliu/asp-coref-t0-3b", device_map="auto")This model is initialized from flan-t5-base and finetuned for coreference resolution task. The model structure is described in the paper Autoregressive Structured Prediction with Language Models, Github repo.
CUDA_VISIBLE_DEVICES=0 python evaluate_coref.py t0_3b tliu/asp-coref-t0-3b 0