File size: 739 Bytes
9e4fe07 d47fdc7 7e8212e d47fdc7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
---
license: mit
---
# RandomPrompt-v1
A fine tuned GPT-neo 125M
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.
## Usage
THE HOSTED INTERFACE DOES NOT WORK, USE THE HUGGINGFACE SPACE
### Autocompletion
Type in a few tags, and it will generate a completion of the prompt
### Generation
Type in nothing, and it will generate a prompt
## Training
Trained on 400k tags from danbooru posts for 600k steps, or around 0.25 epochs
https://wandb.ai/saltacc/RandomPrompt/runs/2v2arf0u?workspace=user-saltacc
I plan on doing further runs on better hardware to try to get more accurate prompt completion
|