kohya-ss / sd-scripts

Apache License 2.0
4.95k stars 829 forks source link

Support new tagger Z3D-E621-Convnext #1257

Open forestsource opened 5 months ago

forestsource commented 5 months ago

Currently support models like the swinv2 tagger, but since some pony models use e621 style tags. I would like support for the Z3D-E621-Convnext model.

kohya-ss commented 5 months ago

The repo have tags-selected.csv but doesn't have selected_tags.csv which is used in the tagger script. And the csv seems to have another format. Is the e621 format polular and are there any other repos with the same format? It is a bit hard work to support only one repo.