From 862f6f996e721c3b1db10066743ebdeef89f592a Mon Sep 17 00:00:00 2001
From: Persiapinkman <1139090915@qq.com>
Date: Thu, 22 May 2025 09:57:48 +0800
Subject: [PATCH 1/2] fix wd14tagger download problem

---
 wd14tagger.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/wd14tagger.py b/wd14tagger.py
index c88f583..1b02d4c 100644
--- a/wd14tagger.py
+++ b/wd14tagger.py
@@ -30,12 +30,9 @@
 }
 defaults.update(config.get("settings", {}))
 
-if "wd14_tagger" in folder_paths.folder_names_and_paths:
-    models_dir = folder_paths.get_folder_paths("wd14_tagger")[0]
-    if not os.path.exists(models_dir):
-        os.makedirs(models_dir)
-else:
-    models_dir = get_ext_dir("models", mkdir=True)
+WD14Tagger_DIR = os.path.join(folder_paths.models_dir, "wd14_tagger")
+models_dir = WD14Tagger_DIR
+
 known_models = list(config["models"].keys())
 
 log("Available ORT providers: " + ", ".join(ort.get_available_providers()), "DEBUG", True)

From f8a61d17c929ca8396a8458354ad167277cbc9da Mon Sep 17 00:00:00 2001
From: Persiapinkman <1139090915@qq.com>
Date: Fri, 30 May 2025 17:12:47 +0800
Subject: [PATCH 2/2] refine requirements of onnx-gpu

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 51decf8..d383298 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1 @@
-onnxruntime
\ No newline at end of file
+onnxruntime-gpu==1.22.0
\ No newline at end of file