flan/winnow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

automated update: 2026-06-10 23:50:06

Browse Source
This commit is contained in:
root
2026-06-10 23:50:06 -04:00
parent 4670ad2646
commit 7210df27c8
6 changed files with 296 additions and 475 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+13 -37
View File
@@ -17,12 +17,15 @@ dependencies = [
"croniter>=5.0.2",
"insightface>=0.7.3",
"numpy>=2.2.6",
"onnxruntime>=1.23.2",
"onnxruntime-gpu>=1.23.2",
"opencv-python-headless>=4.12.0.88",
"pillow>=12.1.0",
"python-dotenv>=1.2.1",
"requests>=2.32.5",
"rich>=14.2.0",
"torch>=2.6.0",
"transformers>=4.57.6",
"ultralytics>=8.3.252",
]
[project.scripts]
@@ -31,49 +34,23 @@ if-curator = "if_curator.cli:main"
[project.urls]
Repository = "https://github.com/sudolulo/if_curator_headless"
[project.optional-dependencies]
cpu = [
"torch>=2.6.0",
]
gpu = [
"onnxruntime-gpu>=1.23.2",
"torch>=2.6.0",
]
object = [
"transformers>=4.57.6",
"ultralytics>=8.3.252",
]
# ── Declare mutually exclusive extras ──
# cpu and gpu cannot be enabled together (they pull torch from different indexes)
# object can be combined with either: --extra cpu --extra object OR --extra gpu --extra object
[tool.uv]
conflicts = [
[
{ extra = "cpu" },
{ extra = "gpu" },
],
]
# ── Per-extra PyTorch index configuration ──
# --extra cpu → CPU-only torch
# --extra gpu → CUDA 12.6 torch
[tool.uv.sources]
torch = [
{ index = "pytorch-cpu", extra = "cpu" },
{ index = "pytorch-cu126", extra = "gpu" },
{ index = "pytorch-cu126", marker = "sys_platform == 'linux' and platform_machine == 'x86_64'" },
{ index = "pytorch-cpu", marker = "sys_platform == 'linux' and platform_machine == 'aarch64'" },
{ index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
]
[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true
[[tool.uv.index]]
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true
[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true
[dependency-groups]
dev = [
"pytest>=8.0",
@@ -99,7 +76,7 @@ opencv-python-headless = "cv2"
python-dotenv = "dotenv"
insightface = "insightface"
numpy = "numpy"
onnxruntime = "onnxruntime"
onnxruntime-gpu = "onnxruntime"
requests = "requests"
rich = "rich"
torch = "torch"
@@ -107,7 +84,6 @@ transformers = "transformers"
ultralytics = "ultralytics"
[tool.deptry.per_rule_ignores]
# onnxruntime-gpu is an optional dep that replaces onnxruntime at runtime
DEP002 = ["onnxruntime-gpu"]
[build-system]