Downloads · 30 days
5
1% of all-time downloads
pegasus_QR
nichal/pegasus_QR
pegasus_QR is a machine learning model from nichal. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
PEGASUS model fine-tuned for the conversational question rewriting task.
All-time downloads
468
Public
Repo size
4.6 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin2.3 GB · 100%
At a glance
- Library
- transformers
- Model type
- pegasus
- Access
- Public
- Created
- Oct 28, 2022
- Updated
- Nov 3, 2022
- SHA
- 71a36114
Model card
From the Hugging Face model README
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.
- Library
- transformers
- Type
- pegasus
- Created
- Oct 28, 2022
- Updated
- Nov 3, 2022