Allows for torch 2.6+

thewhaleking · thewhaleking · commit 9cd2b1c0dd08 · 2025-04-10T17:48:42.000+02:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,7 +39,7 @@ dependencies = [
 
 [project.optional-dependencies]
 cuda = [
-    "torch>=1.13.1,<2.6.0",
+    "torch>=1.13.1,<3.0",
 ]
 
 [project.urls]

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ dependencies = [`
`39`	`39`
`40`	`40`	`[project.optional-dependencies]`
`41`	`41`	`cuda = [`
`42`		`- "torch>=1.13.1,<2.6.0",`
	`42`	`+ "torch>=1.13.1,<3.0",`
`43`	`43`	`]`
`44`	`44`
`45`	`45`	`[project.urls]`