diff --git a/requirements.txt b/requirements.txt
index 51decf8..a3cb77d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1 @@
-onnxruntime
\ No newline at end of file
+onnxruntime-gpu==1.19.2
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)