nichal/pegasus_QR

PEGASUS for Conversational Question Rewriting

Model description

PEGASUS model fine-tuned for the conversational question rewriting task.

Intended uses

Used to rewrite questions using previous conversation turns into self-contained questions (i.e. questions that can be understood outside the conversation context)

How to use

>>> from transformers import PegasusForConditionalGeneration, PegasusTokenizer
>>> tokenizer = PegasusTokenizer.from_pretrained('nichal/pegasus_QR')
>>> model = PegasusForConditionalGeneration.from_pretrained('nichal/pegasus_QR')

Training data

Initialized with pre-trained model PEGASUS-large, fine-tuned using cross-entropy and self-critical sequence training.