diff --git a/install.py b/install.py index 7f445aa..df1fad6 100644 --- a/install.py +++ b/install.py @@ -1,4 +1,4 @@ import launch if not launch.is_installed("onnxruntime-gpu"): - launch.run_pip("install onnxruntime-gpu", "requirements for using SmilingWolf/wd-v1-4-vit-tagger") \ No newline at end of file + launch.run_pip("install onnxruntime-gpu", "requirements for dataset-tag-editor [onnxruntime-gpu]") \ No newline at end of file