diff --git a/requirements/pytorch-lightning.txt b/requirements/pytorch-lightning.txt index 32a0275..7fcd58e 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.2.0 diff --git a/requirements/typing.txt b/requirements/typing.txt index b6cc335..9ccd3ec 100644 --- a/requirements/typing.txt +++ b/requirements/typing.txt @@ -1,2 +1,2 @@ mypy ==1.6.1 -pytorch-lightning ==2.0.9 +pytorch-lightning ==2.1.0