tagger

Akegarasu · Aug 31, 2024 · 931392e · 931392e
1 parent e319e7c
commit 931392e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/mikazuki/tagger/interrogator.py b/mikazuki/tagger/interrogator.py
@@ -231,6 +231,14 @@ def interrogate(
         repo_id='SmilingWolf/wd-v1-4-moat-tagger-v2',
         revision='v2.0'
     ),
+    'wd-eva02-large-tagger-v3': WaifuDiffusionInterrogator(
+        'wd-eva02-large-tagger-v3',
+        repo_id='SmilingWolf/wd-eva02-large-tagger-v3',
+    ),
+    'wd-vit-large-tagger-v3': WaifuDiffusionInterrogator(
+        'wd-vit-large-tagger-v3',
+        repo_id='SmilingWolf/wd-vit-large-tagger-v3',
+    ),
 }