diff --git a/setup.py b/setup.py index c87feaf05e..8828a024cc 100644 --- a/setup.py +++ b/setup.py @@ -78,8 +78,8 @@ def package_files(prefix: str, directory: str, extension: str): 'tqdm>=4.62.3,<5', 'torchmetrics>=0.10.0,<1.3.1', 'torch_optimizer>=0.3.0,<0.4', - 'torchvision>=0.13.1,<0.20', # TODO: Tighten before release - 'torch>=2.0.1,<2.3.1', # TODO: Tighten before release + 'torchvision>=0.13.1,<0.18', + 'torch>=2.0.1,<2.3.0', 'requests>=2.26.0,<3', 'numpy>=1.21.5,<1.27.0', 'psutil>=5.8.0,<6',