Add support smilingwolf's large models.

2024-08-24 11:55:31 +09:00
parent da8baf8d8f
commit d07a80b848
1 changed files with 8 additions and 0 deletions
@@ -45,6 +45,14 @@ interrogators: Dict[str, Interrogator] = {
        'WD14 SwinV2 v3',
        repo_id='SmilingWolf/wd-swinv2-tagger-v3'
    ),
+    'wd-vit-large-tagger-v3': WaifuDiffusionInterrogator(
+        'WD ViT-Large Tagger v3',
+        repo_id='SmilingWolf/wd-vit-large-tagger-v3'
+    ),
+    'wd-eva02-large-tagger-v3': WaifuDiffusionInterrogator(
+        'WD EVA02-Large Tagger v3',
+        repo_id='SmilingWolf/wd-eva02-large-tagger-v3'
+    ),
    'mld-caformer.dec-5-97527': MLDanbooruInterrogator(
        'ML-Danbooru Caformer dec-5-97527',
        repo_id='deepghs/ml-danbooru-onnx',