From 410e79dc6e2c85d0f81b71f33a3870dc930ee9b5 Mon Sep 17 00:00:00 2001 From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com> Date: Tue, 28 Jan 2025 11:32:35 -0500 Subject: [PATCH] Fix #1490 (#1496) --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 267c5e809..515d90385 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -42,7 +42,7 @@ classifiers = [ "Topic :: Scientific/Engineering :: Artificial Intelligence" ] dependencies = [ - "torch~=2.0", + "torch>=2.0,<3", "numpy>=1.17" ]