Edit model card

Disclaimer

Danbooru stores millions of tagged anime images, but it doesn't have a way to filter out NSFW content. This model was trained on 100,000 of these tags with up_score ≥ 3 for 3 epochs, so it's possible that some tags might contain NSFW descriptions. So, just be mindful of that. Thank you for your cooperation.

The Safe Version

For details on data preprocessing, prompt engineering, and more, please see Fast Anime PromptGen. I used a very similar approach to train the Danbooru version.

Downloads last month
66
Inference Examples
Inference API (serverless) has been turned off for this model.

Dataset used to train FredZhang7/danbooru-tag-generator