diff --git a/README.md b/README.md index 33ae9a6..dec6666 100644 --- a/README.md +++ b/README.md @@ -47,16 +47,27 @@ python run.py --dir dir/dir ## Support Models ``` -python run.py --file image.jpg --model wd14-vit.v1 -python run.py --file image.jpg --model wd14-vit.v2 -python run.py --file image.jpg --model wd14-convnext.v1 -python run.py --file image.jpg --model wd14-convnext.v2 -python run.py --file image.jpg --model wd14-convnextv2.v1 -python run.py --file image.jpg --model wd14-swinv2-v1 -python run.py --file image.jpg --model wd-v1-4-moat-tagger.v2 +# SmilingWolf large model. (released 2024) +python run.py --file image.jpg --model wd-vit-large-tagger-v3 +python run.py --file image.jpg --model wd-eva02-large-tagger-v3 + +# SmilingWolf v3 model. (released 2024) python run.py --file image.jpg --model wd-v1-4-vit-tagger.v3 python run.py --file image.jpg --model wd-v1-4-convnext-tagger.v3 python run.py --file image.jpg --model wd-v1-4-swinv2-tagger.v3 + +# SmilingWolf v2 model. (released 2023) +python run.py --file image.jpg --model wd-v1-4-moat-tagger.v2 +python run.py --file image.jpg --model wd14-vit.v2 +python run.py --file image.jpg --model wd14-convnext.v2 + +# SmilingWolf v1 model. (released 2022) +python run.py --file image.jpg --model wd14-vit.v1 +python run.py --file image.jpg --model wd14-convnext.v1 +python run.py --file image.jpg --model wd14-convnextv2.v1 +python run.py --file image.jpg --model wd14-swinv2-v1 + +# kiriyamaX model. python run.py --file image.jpg --model mld-caformer.dec-5-97527 python run.py --file image.jpg --model mld-tresnetd.6-30000 ``` 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',