update wd-v1-4-convnextv2-tagger-v2

2023-03-23 17:29:27 +08:00 · 2023-03-23 17:29:27 +08:00 · 77d4e3569a
parent 590c9a73fb
commit 77d4e3569a
1 changed files with 5 additions and 0 deletions
--- a/tagger/utils.py
+++ b/tagger/utils.py
@ -16,6 +16,11 @@ interrogators: Dict[str, Interrogator] = {}
 def refresh_interrogators() -> List[str]:
    global interrogators
    interrogators = {
+        'wd14-convnextv2-v2': WaifuDiffusionInterrogator(
+            'wd14-convnext-v2',
+            repo_id='SmilingWolf/wd-v1-4-convnextv2-tagger-v2',
+            revision='v3.0'
+        ),
        'wd14-vit-v2': WaifuDiffusionInterrogator(
            'wd14-vit-v2',
            repo_id='SmilingWolf/wd-v1-4-vit-tagger-v2',