diff --git a/setup.py b/setup.py index 8b43d4f2d3..412d831592 100644 --- a/setup.py +++ b/setup.py @@ -78,7 +78,7 @@ def package_files(prefix: str, directory: str, extension: str): 'torchmetrics>=0.10.0,<1.1', 'torch_optimizer>=0.3.0,<0.4', 'torchvision>=0.13.1,<0.17', - 'torch>=1.13.1,<2.1.3', + 'torch>=1.13.1,<2.2.1', 'requests>=2.26.0,<3', 'numpy>=1.21.5,<1.27.0', 'psutil>=5.8.0,<6',