From 1626ceaeb0611dd0c4439948072577ca05152f6e Mon Sep 17 00:00:00 2001 From: Andrew Lapp Date: Tue, 6 Feb 2024 08:00:55 -0600 Subject: [PATCH] simplify torch version range (and disable pytorch<2) --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index f6e29a82e..2505a567d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,7 +32,7 @@ dependencies = [ "diskcache", "pydantic>=2.0", "scipy", - "torch>=1.13.1,!=2.0.*", + "torch>=2.1.0", "numba", "joblib", "referencing",