ros/nanochat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add both sides of the source check

Browse Source
This commit is contained in:
burtenshaw
2025-10-20 10:44:07 +02:00
parent c7ae920a77
commit 0abb0fa2e3
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pyproject.toml
View File

@@ -47,9 +47,15 @@ python_functions = ["test_*"]
# target torch to cuda 12.8 # target torch to cuda 12.8
[tool.uv.sources] [tool.uv.sources]
torch = [ torch = [
{ index = "pytorch-cu128", marker = "platform_system == 'linux'"}, { index = "pytorch-cpu", marker = "sys_platform != 'linux'" },
{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
] ]
[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true
[[tool.uv.index]] [[tool.uv.index]]
name = "pytorch-cu128" name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128" url = "https://download.pytorch.org/whl/cu128"