refix

cfreude · Aug 13, 2022 · 8451ba5 · 8451ba5
1 parent 62238f5
commit 8451ba5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -27,6 +27,7 @@ and activated with:
 ```
 conda env create -f environment.yaml
 conda activate ldm
+conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
 ```
 
 You can also update an existing [latent diffusion](https://github.com/CompVis/latent-diffusion) environment by running

diff --git a/environment.yaml b/environment.yaml
@@ -5,23 +5,25 @@ channels:
 dependencies:
   - python=3.8.5
   - pip=20.3
-  - cudatoolkit=11.3
-  - pytorch=1.11.0
-  - torchvision=0.12.0
+  - cudatoolkit=11.0
+  - pytorch=1.7.0
+  - torchvision=0.8.1
   - numpy=1.19.2
   - pip:
     - albumentations==0.4.3
     - opencv-python==4.1.2.30
     - pudb==2019.2
     - imageio==2.9.0
     - imageio-ffmpeg==0.4.2
-    - pytorch-lightning==1.4.2
+    - pytorch-lightning==1.5.10 
     - omegaconf==2.1.1
     - test-tube>=0.7.5
     - streamlit>=0.73.1
     - einops==0.3.0
     - torch-fidelity==0.3.0
-    - transformers==4.19.2
+    - transformers
+    - webdataset==0.2.5
+    - 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
     - -e .