Downloads · 30 days
13
2% of all-time downloads
opt-125-imdb-ppo
mwitiderrick/opt-125-imdb-ppo
opt-125-imdb-ppo is a text generation model from mwitiderrick. Use it when you need the model to write or continue text. It is set up for transformers.
We fine-tune OPT-125 to generate positive movie reviews based on the IMDB dataset. The model gets the start of a real review and is tasked to produce positive continuations. To reward positive continuations we use a B…
All-time downloads
542
Public
Parameters
125M
1 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors501 MB · 99%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- opt
- Access
- Public
- Created
- Jul 10, 2024
- Updated
- Jul 11, 2024
- SHA
- 6293cb8d
Try a prompt
Model card
From the Hugging Face model README
Model Card for Model ID
We fine-tune OPT-125 to generate positive movie reviews based on the IMDB dataset. The model gets the start of a real review and is tasked to produce positive continuations. To reward positive continuations we use a BERT classifier to analyse the sentiment of the produced sentences and use the classifier's outputs as rewards signals for PPO training
- Task
- Text Generation
- Library
- transformers
- Type
- opt
- Created
- Jul 10, 2024
- Updated
- Jul 11, 2024