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',