pavlichenko commited on
Commit
8cfae50
1 Parent(s): 11bcaea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -0
README.md CHANGED
@@ -1,3 +1,22 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ language:
4
+ - en
5
+ library_name: transformers
6
+ pipeline_tag: text-generation
7
+ tags:
8
+ - rlhf
9
+ - stable diffusion
10
  ---
11
+
12
+ # GPT2-Large For Writing Stable Diffusion Prompts
13
+
14
+ 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.
15
+
16
+ ## How To Use a Model
17
+
18
+ Write an image description and add a special SEP token: `</s>`, the model writes the prompt after it. Example:
19
+
20
+ ```
21
+ a cat</s>
22
+ ```