Downloads · 30 days
17
2% of all-time downloads
gpt2-large-rl-prompt-writing
toloka/gpt2-large-rl-prompt-writing
gpt2-large-rl-prompt-writing is a text generation model from toloka. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
This is a GPT2-Large model fine-tuned for rewriting prompts for Stable Diffusion v1.5 through RLHF. The model takes an image description and writes a prompt resulting in aestheticcally pleasing images.
All-time downloads
854
Public
Parameters
815M
6.3 GB on disk
Likes
3
Public
Repo makeup
Click a slice to open those files.
.bin3.1 GB · 50%
Parameter types
How the weights are stored.
F32777M · 95%
Try a prompt
Model card
From the Hugging Face model README
GPT2-Large For Writing Stable Diffusion Prompts
This is a GPT2-Large model fine-tuned for rewriting prompts for Stable Diffusion v1.5 through RLHF. The model takes an image description and writes a prompt resulting in aestheticcally pleasing images.
How To Use a Model
Write an image description and add a special SEP token: </s>, the model writes the prompt after it. Example:
a cat</s>
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- License
- mit
- Languages
- en
- Created
- Apr 10, 2023
- Updated
- Apr 21, 2023