From d8bbbce5d54b419bf62d5752a6e9d6dae3e0003a Mon Sep 17 00:00:00 2001 From: narugo1992 Date: Sun, 28 Jul 2024 05:40:50 +0800 Subject: [PATCH] dev(narugo): add vit large model into list --- imgutils/tagging/wd14.py | 2 +- zoo/wd14/sync.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/imgutils/tagging/wd14.py b/imgutils/tagging/wd14.py index fe1696dafa..74db08f9ce 100644 --- a/imgutils/tagging/wd14.py +++ b/imgutils/tagging/wd14.py @@ -43,7 +43,7 @@ "ConvNext_v3": CONV_V3_MODEL_REPO, "ViT_v3": VIT_V3_MODEL_REPO, - "Vit_Large": VIT_LARGE_MODEL_REPO, + "ViT_Large": VIT_LARGE_MODEL_REPO, } _DEFAULT_MODEL_NAME = 'SwinV2_v3' diff --git a/zoo/wd14/sync.py b/zoo/wd14/sync.py index d664b95c6b..92f30458c8 100644 --- a/zoo/wd14/sync.py +++ b/zoo/wd14/sync.py @@ -47,6 +47,7 @@ def _seg_split(text): "SwinV2_v3": ('core_model', 'head', 'fc'), "ConvNext_v3": ('core_model', 'head', 'fc'), "ViT_v3": ('core_model', 'head'), + "ViT_Large": ('core_model', 'head'), }