[project.optional-dependencies]
cpu = [
"torch==2.8.0+cpu",
"torchvision==0.23.0+cpu",
"torchaudio==2.8.0+cpu",
]
cu128 = [
"torch==2.8.0+cu128",
"torchvision==0.23.0+cu128",
"torchaudio==2.8.0+cu128",
]
[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true
[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true
[tool.uv.sources]
torch = [
{ index = "pytorch-cpu", extra = "cpu" },
{ index = "pytorch-cu128", extra = "cu128" },
]
torchvision = [
{ index = "pytorch-cpu", extra = "cpu" },
{ index = "pytorch-cu128", extra = "cu128" },
]
[tool.uv]
index-url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
conflicts = [[
{ extra = "cpu" },
{ extra = "cu128" },
]]