diff --git a/requirements/base.txt b/requirements/base.txt index ce30aff1..97049a64 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -2,7 +2,7 @@ torch >=1.11.0 numpy >1.20.0, <2.0 pandas >=1.1.5 scikit-learn >=1.3.0 -pytorch-lightning >=2.0.0, <2.5.0 +pytorch-lightning >=2.0.0, <2.6.0 omegaconf >=2.3.0 torchmetrics >=0.10.0, <1.7.0 tensorboard >2.2.0, !=2.5.0