From 09fd11ae88bb2428763dc24d574e87d7f38846be Mon Sep 17 00:00:00 2001
From: narugo1992 <narugo992@gmail.com>
Date: Sun, 28 Jul 2024 05:45:27 +0800
Subject: [PATCH] dev(narugo): add vit large model into list

---
 imgutils/tagging/wd14.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/imgutils/tagging/wd14.py b/imgutils/tagging/wd14.py
index 74db08f9ce..fbd716571f 100644
--- 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'