From 9cd2b1c0dd0892e879d28c3d02a515f2ca2039e7 Mon Sep 17 00:00:00 2001 From: Benjamin Himes Date: Thu, 10 Apr 2025 17:48:42 +0200 Subject: [PATCH] Allows for torch 2.6+ --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 840807700..37cfd4aeb 100644 --- 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]