Merge pull request #198 from MapleWithered/patch-1

fix: fix triton source from default cuda129 to 126
This commit is contained in:
小铭
2025-08-27 21:31:46 +08:00
committed by GitHub
+4
View File
@@ -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]]