diff --git a/requirements/requirements-pytorch.txt b/requirements/requirements-pytorch.txt index 03f4e997ab..683bb101d7 100644 --- a/requirements/requirements-pytorch.txt +++ b/requirements/requirements-pytorch.txt @@ -1,9 +1,6 @@ torch>=1.9,<3 -lightning>=1.8,<2.2 +lightning>=2.0,<2.2 # Capping `lightning` does not cap `pytorch_lightning`, so we cap manually -pytorch_lightning>=1.8,<2.2 -# Need to pin protobuf (for now) -# See: https://github.com/PyTorchLightning/pytorch-lightning/issues/13159 -protobuf~=3.19.0 +pytorch_lightning>=2.0,<2.2 scipy~=1.10; python_version > "3.7.0" -scipy~=1.7.3; python_version <= "3.7.0" \ No newline at end of file +scipy~=1.7.3; python_version <= "3.7.0"