From 85667ca6767cce14b343c911827fc79f3a3d331e Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Wed, 25 Mar 2026 13:22:24 +0000 Subject: [PATCH] Add cu126 sources --- pyproject.toml | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index c599887..7da7d2f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -60,6 +60,22 @@ dev = [ "ruff>=0.7.3", ] +[[tool.uv.index]] +name = "pytorch-cu126" +url = "https://download.pytorch.org/whl/cu126" +explicit = true + +[tool.uv.sources] +torch = [ + { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }, +] +torchvision = [ + { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }, +] +torchaudio = [ + { index = "pytorch-cu126", marker = "sys_platform == 'linux' or sys_platform == 'win32'" }, +] + [tool.ruff] line-length = 79 target-version = "py39"