picobyte / stable-diffusion-webui-wd14-tagger

Labeling extension for Automatic1111's Web UI
539 stars 64 forks source link

Support for Smiling Wolf new V3 models #100

Open trihardseven opened 3 months ago

trihardseven commented 3 months ago

Requesting support for these new models as manual downloading didn't work for me: https://huggingface.co/SmilingWolf/wd-vit-tagger-v3 https://huggingface.co/SmilingWolf/wd-convnext-tagger-v3 https://huggingface.co/SmilingWolf/wd-swinv2-tagger-v3

songkunpeng52038 commented 3 months ago

Have you attempted to edit a file that named utils.py in "stable-diffusion-webui\extensions\stable-diffusion-webui-wd14-tagger\tagger"? like this, just add it behind that ones 屏幕截图 2024-04-01 141454

GiusTex commented 2 months ago

Have you attempted to edit a file that named utils.py in "stable-diffusion-webui\extensions\stable-diffusion-webui-wd14-tagger\tagger"?

You need also to have an updated version of onnx, as stated here. I too changed the script to use the v3 versions and got Error completing request, then in the auto1111 venv I reinstalled the onnx library and it worked