Merge pull request #198 from MapleWithered/patch-1

xming521 · web-flow · commit 7a5bcb491eae · 2025-08-27T21:31:46.000+08:00
fix: fix triton source from default cuda129 to 126
diff --git 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]]

Original file line number	Diff line number	Diff line change
`@@ -79,6 +79,10 @@ torchvision = [`
`79`	`79`	`{ index = "pytorch-cu126", marker = "platform_system == 'Windows'" },`
`80`	`80`	`{ index = "pytorch-cu126", marker = "platform_system == 'Linux'" },`
`81`	`81`	`]`
	`82`	`+triton = [`
	`83`	`+ { index = "pytorch-cu126", marker = "platform_system == 'Windows'" },`
	`84`	`+ { index = "pytorch-cu126", marker = "platform_system == 'Linux'" },`
	`85`	`+]`
`82`	`86`
`83`	`87`
`84`	`88`	`[[tool.uv.index]]`