search

theovercomer8 / captionr

GIT/BLIP/CLIP Caption tool
MIT License
138 stars 15 forks source link

Add model based on DeepDanbooru #11

Open MitPitt opened 1 year ago

MitPitt commented 1 year ago

DeepDanbooru is a caption model trained on tags from boorus. Mostly anime, but it does work incredibly well with non-anime images too. Booru-style prompts (example: 1girl, simple background, watercolor, ...) work very well in text2img.

DeepDanbooru is originally used on danbooru to tag images, but now is also used to fine-tune SD. Since then a couple modifications (like removing underscores in tags) have been made to it, to work better with SD.

See demo and links to models here: https://huggingface.co/spaces/SmilingWolf/wd-v1-4-tags

theovercomer8 commented 1 year ago

Will definitely consider doing soon.