search

pythongosssss / ComfyUI-WD14-Tagger

A ComfyUI extension allowing for the interrogation of booru tags from images.
MIT License
436 stars 51 forks source link

[Feature Request] Add SwinV2 model #24

Closed unLomTrois closed 3 months ago

unLomTrois commented 4 months ago

Nice extension, any chance to add swinv2 model?

https://huggingface.co/SmilingWolf/wd-v1-4-swinv2-tagger-v2

camoody1 commented 3 months ago

@unLomTrois I just made two Pull Requests with the necessary code changes to add the swinv2 tagger model. You can check those requests to see the changes if you want to make them yourself. They're super short and simple.