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" ]