From 77d4e3569a6858d4f1c50c7a6ebde43fef02af91 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9D=92=E9=BE=8D=E8=81=96=E8=80=85=40bdsqlsz?=
 <qinglongshengzhe@gmail.com>
Date: Thu, 23 Mar 2023 17:29:27 +0800
Subject: [PATCH] update wd-v1-4-convnextv2-tagger-v2

---
 tagger/utils.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tagger/utils.py b/tagger/utils.py
index 21f1135..d640cd3 100644
--- 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',