metadata

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>