diff --git a/setup.py b/setup.py index 97772a0..73c5ea6 100644 --- a/setup.py +++ b/setup.py @@ -23,7 +23,7 @@ exec(f.read(), version_file) install_requires = [ - "torch>=1.2,<2.1; sys_platform != 'win32'", + "torch>=2.5,<3; sys_platform != 'win32'", "tqdm>=4.23", "numpy>=1.13", "scipy>=1.3.3",