From c384987b8a30f37bdc830c8bd9a8680c93262bfb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9E=AB=E4=BA=9A?= <47618005+MapleWithered@users.noreply.github.com> Date: Wed, 27 Aug 2025 19:42:48 +0800 Subject: [PATCH] fix: fix triton source from cuda129 to 126 --- pyproject.toml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index f998a30..c5525b0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -79,6 +79,10 @@ torchvision = [ { index = "pytorch-cu126", marker = "platform_system == 'Windows'" }, { index = "pytorch-cu126", marker = "platform_system == 'Linux'" }, ] +triton = [ + { index = "pytorch-cu126", marker = "platform_system == 'Windows'" }, + { index = "pytorch-cu126", marker = "platform_system == 'Linux'" }, +] [[tool.uv.index]]