diff --git a/requirements/requirements-pytorch.txt b/requirements/requirements-pytorch.txt index 9947b4e9ae..03f4e997ab 100644 --- a/requirements/requirements-pytorch.txt +++ b/requirements/requirements-pytorch.txt @@ -1,6 +1,7 @@ torch>=1.9,<3 lightning>=1.8,<2.2 -lightning.pytorch>=1.8,<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