Downloads · 30 days
15
3% of all-time downloads
DialoGPT-small-PFG
LACAI/DialoGPT-small-PFG
DialoGPT-small-PFG is a text generation model from LACAI. Use it when you need the model to write or continue text. It is set up for transformers.
All-time downloads
503
Public
Repo size
1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin510 MB · 99%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jan 14, 2022
- SHA
- 905128cb
Try a prompt
Model card
From the Hugging Face model README
Base model: microsoft/DialoGPT-small
Fine tuned for dialogue response generation on the Persuasion For Good Dataset (Wang et al., 2019)
Three additional special tokens were added during the fine-tuning process:
- <|pad|> padding token
- <|user|> speaker control token to prompt user responses
- <|system|> speaker control token to prompt system responses
The following Dialogues were excluded:
- Those with donation amounts outside of the task range of [$0, $2].
- Those where a donation of 0 was made at the end of the task but a non-zero amount was pledged in the dialogue.
- Those with more than 800 words.
Stats:
- Training set: 519 dialogues
- Validation set: 58 dialogues
- ~20 utterances per dialogue
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- Created
- Mar 2, 2022
- Updated
- Jan 14, 2022