dev(narugo): add vit large model into list

deepghs · Jul 27, 2024 · 09fd11a · 09fd11a
1 parent d8bbbce
commit 09fd11a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/imgutils/tagging/wd14.py b/imgutils/tagging/wd14.py
@@ -33,6 +33,8 @@
 _IS_V3_SUPPORT = VersionInfo(onnxruntime.__version__) >= '1.17'
 
 MODEL_NAMES = {
+    "ViT_Large": VIT_LARGE_MODEL_REPO,
+
     "SwinV2": SWIN_MODEL_REPO,
     "ConvNext": CONV_MODEL_REPO,
     "ConvNextV2": CONV2_MODEL_REPO,
@@ -42,8 +44,6 @@
     "SwinV2_v3": SWIN_V3_MODEL_REPO,
     "ConvNext_v3": CONV_V3_MODEL_REPO,
     "ViT_v3": VIT_V3_MODEL_REPO,
-
-    "ViT_Large": VIT_LARGE_MODEL_REPO,
 }
 _DEFAULT_MODEL_NAME = 'SwinV2_v3'