diff --git a/requirements/pytorch-lightning.txt b/requirements/pytorch-lightning.txt index 32a0275..60c5952 100644 --- a/requirements/pytorch-lightning.txt +++ b/requirements/pytorch-lightning.txt @@ -1,4 +1,4 @@ # this sets the requirements contains if you go with main lightning # in 2.0.7 we have removed pytorch_lightning.overrides.base._LightningPrecisionModuleWrapperBase -pytorch-lightning >=2.0.0, <=2.0.6 +pytorch-lightning >=2.0.0, <=2.1.0 diff --git a/requirements/typing.txt b/requirements/typing.txt index c589e6d..151ea92 100644 --- a/requirements/typing.txt +++ b/requirements/typing.txt @@ -1,2 +1,2 @@ mypy ==1.5.1 -pytorch-lightning ==2.0.9 +pytorch-lightning ==2.1.0