dev(narugo): add vit large model into list

deepghs · Jul 27, 2024 · d8bbbce · d8bbbce
1 parent f886d62
commit d8bbbce
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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'),
 }