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.