From ae00f1ff8b156141f6aec1c75ea72f90ce49f02b Mon Sep 17 00:00:00 2001
From: aria1th <35677394+aria1th@users.noreply.github.com>
Date: Wed, 7 May 2025 15:00:06 +0900
Subject: [PATCH] How

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

diff --git a/utils/tagger.py b/utils/tagger.py
index d9feda3..f609feb 100644
--- a/utils/tagger.py
+++ b/utils/tagger.py
@@ -4,6 +4,18 @@ try:
 except ImportError:
     def get_wd14_tags(image_path):
         raise Exception("Tagger feature not available, please install dghs-imgutils")
+    tagger_model_names = {
+        "EVA02_Large": None,
+        "ViT_Large": None,
+        "SwinV2": None,
+        "ConvNext": None,
+        "ConvNextV2": None,
+        "ViT": None,
+        "MOAT": None,
+        "SwinV2_v3": None,
+        "ConvNext_v3": None,
+        "ViT_v3": None,
+    }
 from typing import Union
 from PIL import Image