diff --git a/README.md b/README.md index 4057293..b57402f 100644 --- a/README.md +++ b/README.md @@ -84,9 +84,9 @@ 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/ ``` -https://onnxruntime.ai/docs/install/ +https://onnxruntime.ai/docs/install/
https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html#requirements ## 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`) 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 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