Update README.md
Browse files
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 |
+
|