saltacc commited on
Commit
d47fdc7
1 Parent(s): 0802651

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -0
README.md CHANGED
@@ -1,3 +1,29 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+ # RandomPrompt-v1
6
+
7
+ A fine tuned GPT-neo 125M
8
+
9
+ The purpose of this model is to autocomplete or generate danbooru-like prompts for generating images in Stable Diffusion derivatives that use danbooru tags for text conditioning.
10
+
11
+ ## Usage
12
+
13
+ ### Autocompletion
14
+
15
+ Type in a few tags, and it will generate a completion of the prompt
16
+
17
+ ### Generation
18
+
19
+ Type in nothing, and it will generate a prompt
20
+
21
+
22
+ ## Training
23
+
24
+ Trained on 400k tags from danbooru posts for 600k steps, or around 0.25 epochs
25
+
26
+ https://wandb.ai/saltacc/RandomPrompt/runs/2v2arf0u?workspace=user-saltacc
27
+
28
+ I plan on doing further runs on better hardware to try to get more accurate prompt completion
29
+