search

Linaqruf / kohya-trainer

Adapted from https://note.com/kohya_ss/n/nbf7ce8d80f29 for easier cloning
Apache License 2.0
1.82k stars 296 forks source link

Please update the tagger, add V3 tagger #349

Open konokomi opened 3 months ago

konokomi commented 3 months ago

https://huggingface.co/SmilingWolf/wd-vit-tagger-v3 https://huggingface.co/SmilingWolf/wd-swinv2-tagger-v3 https://huggingface.co/SmilingWolf/wd-convnext-tagger-v3

thankyou

JonesHong commented 3 months ago

Go to 4.2.2 eidt code

model = "SmilingWolf/wd-swinv2-tagger-v3" #@param ["SmilingWolf/wd-vit-tagger-v3","SmilingWolf/wd-swinv2-tagger-v3","SmilingWolf/wd-convnext-tagger-v3","SmilingWolf/wd-v1-4-convnextv2-tagger-v2", "SmilingWolf/wd-v1-4-swinv2-tagger-v2", "SmilingWolf/wd-v1-4-convnext-tagger-v2", "SmilingWolf/wd-v1-4-vit-tagger-v2"]

It's all you need!