From b5faf9ded5000ee3f43e3727dd2021c5b94fb647 Mon Sep 17 00:00:00 2001 From: corkborg <62742115+corkborg@users.noreply.github.com> Date: Sat, 1 Feb 2025 00:49:45 +0900 Subject: [PATCH 1/3] Update dependencies. --- README.md | 4 ++-- requirements.txt | 14 +++++++------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index 4057293..b32482f 100644 --- a/README.md +++ b/README.md @@ -78,13 +78,13 @@ python run.py --file image.jpg --model mld-tresnetd.6-30000 ## Using GPU -Requires CUDA 12.2 and cuDNN8.x. +Requires CUDA 12.2 and cuDNN9.x. ``` pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ ``` -https://onnxruntime.ai/docs/install/ +https://onnxruntime.ai/docs/install/
https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements ## Copyright diff --git a/requirements.txt b/requirements.txt index ac86f53..40be636 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ -pandas==2.2.2 -huggingface-hub==0.22.2 -Pillow==10.0.1 -deepdanbooru==1.0.2 -onnxruntime==1.17.3 -numpy==1.26.4 -opencv-python==4.9.0.80 +pandas==2.2.3 +huggingface-hub==0.28.1 +Pillow==11.1.0 +deepdanbooru==1.0.3 +onnxruntime==1.20.1 +numpy==2.2.2 +opencv-python==4.11.0.86 From 30b547cb67dfa8e2844efcf4c4b436c981646309 Mon Sep 17 00:00:00 2001 From: corkborg <62742115+corkborg@users.noreply.github.com> Date: Tue, 25 Mar 2025 18:11:03 +0900 Subject: [PATCH 2/3] Supress warning. --- run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run.py b/run.py index 92aa2ce..df42da4 100644 --- a/run.py +++ b/run.py @@ -68,7 +68,7 @@ def parse_exclude_tags() -> set[str]: # reverse escape (nai tag to danbooru tag) reverse_escaped_tags = [] for tag in tags: - tag = tag.replace(' ', '_').replace('\(', '(').replace('\)', ')') + tag = tag.replace(' ', '_').replace(r'\(', '(').replace(r'\)', ')') reverse_escaped_tags.append(tag) return set([*tags, *reverse_escaped_tags]) # reduce duplicates From fe065d6c05585f8ac2f23d1d7b48c843b2d89692 Mon Sep 17 00:00:00 2001 From: corkborg <62742115+corkborg@users.noreply.github.com> Date: Tue, 25 Mar 2025 18:13:11 +0900 Subject: [PATCH 3/3] Revert readme. --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index b32482f..b57402f 100644 --- a/README.md +++ b/README.md @@ -78,7 +78,7 @@ python run.py --file image.jpg --model mld-tresnetd.6-30000 ## Using GPU -Requires CUDA 12.2 and cuDNN9.x. +Requires CUDA 12.2 and cuDNN8.x. ``` pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ @@ -89,4 +89,4 @@ https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requ ## Copyright -Public domain, except borrowed parts (e.g. `dbimutils.py`) \ No newline at end of file +Public domain, except borrowed parts (e.g. `dbimutils.py`)