gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

torch dep fix

Browse source
This commit is contained in:
phact 2025-09-12 13:11:32 -04:00
parent 463bb48222
commit 364f24a2ca
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pyproject.toml
View file

@ -43,12 +43,12 @@ package = true
[tool.uv.sources]
torch = [
{ index = "pytorch-cu128", extra = "torch-cu128" },
{ index = "pytorch-cpu", extra = "torch" }
{ index = "pytorch-cu128", extra = "torch-cu128", marker = "sys_platform == 'linux'" },
{ index = "pytorch-cpu", extra = "torch", marker = "sys_platform != 'linux'" }
]
torchvision = [
{ index = "pytorch-cu128", extra = "torch-cu128" },
{ index = "pytorch-cpu", extra = "torch" }
{ index = "pytorch-cu128", extra = "torch-cu128", marker = "sys_platform == 'linux'" },
{ index = "pytorch-cpu", extra = "torch", marker = "sys_platform != 'linux'" }
]
[[tool.uv.index]]