chore: add pytorch source

2025-11-12 22:58:43 +09:00
parent d417b2a63f
commit 376766163e
2 changed files with 6 additions and 1 deletions
@@ -2947,4 +2947,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.10,<3.15"
-content-hash = "914eca918159def95f38d1366b6ba0a8146d7fa904b4df45becfbe5745ff8bc2"
+content-hash = "1a98a1675830c27077e77d0ad5f2eb0ef11091534bb7889d1fab6c6d4a6ffed6"
@@ -23,6 +23,11 @@ dependencies = [
 [tool.poetry]
 packages = [{ include = "src" }]

+[[tool.poetry.source]]
+name = "pytorch-cu124"
+url = "https://download.pytorch.org/whl/cu124"
+priority = "explicit"
+
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 python_files = ["test_*.py", "*_test.py"]