This is onnx models based on SmilingWolf's wd14 anime taggers, which added the embeddings output as the second output.
8 models in total:
| Name | Source Repository | Tags Count | Embedding Width | Inverse Supported |
|---|---|---|---|---|
| SwinV2 | SmilingWolf/wd-v1-4-swinv2-tagger-v2 | 9083 | 1024 | Yes |
| ConvNext | SmilingWolf/wd-v1-4-convnext-tagger-v2 | 9083 | 1024 | Yes |
| ConvNextV2 | SmilingWolf/wd-v1-4-convnextv2-tagger-v2 | 9083 | 1024 | Yes |
| ViT | SmilingWolf/wd-v1-4-vit-tagger-v2 | 9083 | 768 | Yes |
| MOAT | SmilingWolf/wd-v1-4-moat-tagger-v2 | 9083 | 1024 | Yes |
| SwinV2_v3 | SmilingWolf/wd-swinv2-tagger-v3 | 10861 | 1024 | Yes |
| ConvNext_v3 | SmilingWolf/wd-convnext-tagger-v3 | 10861 | 1024 | Yes |
| ViT_v3 | SmilingWolf/wd-vit-tagger-v3 | 10861 | 768 | Yes |
Unable to determine this model's library. Check the
docs
.