Downloads · 30 days
299
3% of all-time downloads
FredZhang7/danbooru-tag-generator
danbooru-tag-generator is a text generation model from FredZhang7. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as creativeml-openrail-m.
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 upscore ≥ 3 for 3 epochs, so it's possible that some tags mig…
Downloads · 30 days
299
3% of all-time downloads
All-time downloads
11.6K
Public
Repo size
1.2 GB
Likes
20
Public
Click a slice to open those files.
.onnx483 MB · 59%
From the Hugging Face model README
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.
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.