diff --git a/pyproject.toml b/pyproject.toml index 13447b57..4913dbae 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -43,7 +43,7 @@ dependencies = [ ] [project.optional-dependencies] -gpu = ["torch==2.8.0+cu129"] +gpu = ["torch==2.8.0+cu126"] cpu = ["torch==2.8.0"] test = [ "pytest==8.3.5", @@ -70,7 +70,7 @@ explicit = true [[tool.uv.index]] name = "pytorch-cuda" -url = "https://download.pytorch.org/whl/cu129" +url = "https://download.pytorch.org/whl/cu126" explicit = true [build-system]