toriato / stable-diffusion-webui-wd14-tagger

Labeling extension for Automatic1111's Web UI
1.34k stars 235 forks source link

E621 support #92

Closed picobyte closed 1 year ago

picobyte commented 1 year ago

Add support for E621

credits to Zack. I just applied the changes to the repository from: https://discord.com/channels/754509198674362388/1065785788698218526/1067966541980782712 And the changes: Resolved merge conflicts, newline line endings, in place of \n\r (so the changes are readable) Rebased with toriato current

As Zack designed, the tagger goes in stable-diffusion-webui/models/TaggerOnnx/ and can be found here: https://discord.com/channels/754509198674362388/1065785788698218526/1065785788698218526 (just unzip it there, leave the directory, may require a stable-diffusion-webui restart)

This should fix issue #75

picobyte commented 1 year ago

Similarly this should be able to fix #64, I'll send an update. That will also require to download the repo.

cd stable-diffusion-webui/models/TaggerOnnx/
git clone https://huggingface.co/SmilingWolf/wd-v1-4-convnextv2-tagger-v2/
picobyte commented 1 year ago

It seems that checking 'Include confident of tags matches in results' for an onnx model triggers a bug.