kapitan/wdtagger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(dependencies): add image tagging script && update torch dependency

Browse Source
This commit is contained in:
Jianqi Pan
2025-02-21 15:06:01 +09:00
parent 92ab8bcbcc
commit 458a0d9410
2 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pyproject.toml
View File

@@ -17,8 +17,8 @@ dev = ["pytest>=8.3.3", "pytest-benchmark>=5.1.0", "ruff>=0.8.0"]
[project.optional-dependencies]
cpu = ["torch>=2.5.1", "torchvision>=0.20.1", "timm>=1.0.11"]
gpu = ["torch>=2.5.1", "torchvision>=0.20.1", "timm>=1.0.11"]
cpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
gpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
[tool.ruff]
line-length = 140