fix: revert to CUDA 12.6 base for onnxruntime-gpu compatibility

2026-06-11 10:00:51 -04:00
parent b0ee06ee51
commit dac2bc8242
2 changed files with 199 additions and 249 deletions
@@ -68,5 +68,27 @@ target-version = "py312"
 [tool.ruff.lint]
 select = ["E", "F", "I"]

-[
+[tool.deptry]
+pep621_dev_dependency_groups = ["dev"]
+
+[tool.deptry.package_module_name_map]
+pillow = "PIL"
+opencv-python-headless = "cv2"
+python-dotenv = "dotenv"
+insightface = "insightface"
+numpy = "numpy"
+nvidia-cudnn-cu12 = "nvidia.cudnn"
+onnxruntime-gpu = "onnxruntime"
+requests = "requests"
+rich = "rich"
+torch = "torch"
+transformers = "transformers"
+ultralytics = "ultralytics"
+
+[tool.deptry.per_rule_ignores]
+DEP002 = ["onnxruntime-gpu", "nvidia-cudnn-cu12"]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"