File size: 532 Bytes
d023b86 8cfae50 d023b86 8cfae50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
---
license: mit
language:
- en
library_name: transformers
pipeline_tag: text-generation
tags:
- rlhf
- stable diffusion
---
# 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>
``` |