diff --git a/requirements.txt b/requirements.txt index 9ac8f4f..3590d36 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ numpy>=1.24.4,<2.0.0 omegaconf>=2.3.0 open-clip-torch==2.24.0 pillow>=9.5.0 -pytorch-lightning==1.8.5 +pytorch-lightning==2.4.0 pyyaml>=6.0.1 tokenizers>=0.19,<0.20 --extra-index-url https://download.pytorch.org/whl/cu117