diff --git a/docker/training/requirements-s.txt b/docker/training/requirements-s.txt index 60c50c1..b9d0734 100644 --- a/docker/training/requirements-s.txt +++ b/docker/training/requirements-s.txt @@ -10,7 +10,7 @@ diffusers==0.6.0 huggingface-hub transformers==4.24.0 tqdm==4.64.0 -pytorch-lightning==1.8.1 +pytorch-lightning==1.6.0 numpy -e git+https://github.com/CompVis/taming-transformers.git@24268930bf1dce879235a7fddd0b2355b84d7ea6#egg=taming_transformers diff --git a/docker/training/requirements.txt b/docker/training/requirements.txt index 4ec1b31..910f401 100644 --- a/docker/training/requirements.txt +++ b/docker/training/requirements.txt @@ -71,7 +71,7 @@ Pympler==1.0.1 pyparsing==3.0.9 pyrsistent==0.18.1 python-dateutil==2.8.2 -pytorch-lightning==1.4.2 +pytorch-lightning==1.6.0 pytz==2022.2.1 pytz-deprecation-shim==0.1.0.post0 PyWavelets==1.3.0