diff --git a/environment.yaml b/environment.yaml index 7f25da8..19ddcd8 100644 --- a/environment.yaml +++ b/environment.yaml @@ -3,7 +3,7 @@ channels: - pytorch - defaults dependencies: - - python=3.8.5 + - python=3.8.13 - pip=20.3 - cudatoolkit=11.3 - pytorch=1.11.0 @@ -23,6 +23,8 @@ dependencies: - torch-fidelity==0.3.0 - transformers==4.19.2 - torchmetrics==0.6.0 + - pywavelets==1.3.0 + - pandas==1.4.4 - kornia==0.6 - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers - -e git+https://github.com/openai/CLIP.git@main#egg=clip