diff --git a/multimodal/setup.py b/multimodal/setup.py index 1d8e84039b3..17d887dc64a 100644 --- a/multimodal/setup.py +++ b/multimodal/setup.py @@ -37,7 +37,7 @@ "torchvision>=0.10.0,<0.16", "fairscale>=0.4.5,<0.4.14", "scikit-image>=0.19.1,<0.21.0", - "pytorch-lightning>=1.9.0,<2.1", + "pytorch-lightning>=2.0.0,<2.1", "text-unidecode>=1.3,<1.4", "torchmetrics~=1.0.0rc0", "transformers>=4.23.0,<4.27.0", diff --git a/timeseries/setup.py b/timeseries/setup.py index 52b1d608ecf..57092cfc755 100644 --- a/timeseries/setup.py +++ b/timeseries/setup.py @@ -30,7 +30,7 @@ "statsmodels>=0.13.0,<0.14", "gluonts>=0.12.4,<0.13", "torch>=1.9,<2.1", - "pytorch-lightning>=1.9.0,<2.1", + "pytorch-lightning>=2.0.0,<2.1", "networkx", # version range defined in `core/_setup_utils.py` "statsforecast>=1.4.0,<1.5", "mlforecast>=0.7.0,<0.8.0",