Add wd14 v3 models.

2024-05-18 15:37:12 +09:00
parent b6d257f55d
commit 57304e78e6
3 changed files with 31 additions and 6 deletions
@@ -33,6 +33,18 @@ interrogators: Dict[str, Interrogator] = {
        'WD14 moat tagger v2',
        repo_id='SmilingWolf/wd-v1-4-moat-tagger-v2'
    ),
+    'wd-v1-4-vit-tagger.v3': WaifuDiffusionInterrogator(
+        'WD14 ViT v3',
+        repo_id='SmilingWolf/wd-vit-tagger-v3'
+    ),
+    'wd-v1-4-convnext-tagger.v3': WaifuDiffusionInterrogator(
+        'WD14 ConvNext v3',
+        repo_id='SmilingWolf/wd-convnext-tagger-v3'
+    ),
+    'wd-v1-4-swinv2-tagger.v3': WaifuDiffusionInterrogator(
+        'WD14 SwinV2 v3',
+        repo_id='SmilingWolf/wd-swinv2-tagger-v3'
+    ),
    'mld-caformer.dec-5-97527': MLDanbooruInterrogator(
        'ML-Danbooru Caformer dec-5-97527',
        repo_id='deepghs/ml-danbooru-onnx',