From d07a80b848bcba3863acc86b820b8bea30dd8c36 Mon Sep 17 00:00:00 2001
From: corkborg <62742115+corkborg@users.noreply.github.com>
Date: Sat, 24 Aug 2024 11:55:31 +0900
Subject: [PATCH] Add support smilingwolf's large models.

---
 tagger/interrogators.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/tagger/interrogators.py b/tagger/interrogators.py
index cf447be..13c2859 100644
--- a/tagger/interrogators.py
+++ b/tagger/interrogators.py
@@ -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',