added cpuonly environment script

README.md

@@ -25,16 +25,16 @@ A suitable [conda](https://conda.io/) environment named `ldm` can be created
 and activated with:
 
 ```
-conda env create -f environment.yaml
+conda env create -f environment-cpuonly.yaml
 conda activate ldm
 ```
 
-You can also update an existing [latent diffusion](https://github.com/CompVis/latent-diffusion) environment by running
+//You can also update an existing [latent diffusion](https://github.com/CompVis/latent-diffusion) environment by running
-
+//
-```
+//```
-conda install pytorch torchvision -c pytorch
+//conda install pytorch torchvision -c pytorch
-pip install transformers==4.19.2 diffusers invisible-watermark
+//pip install transformers==4.19.2 diffusers invisible-watermark
-pip install -e .
+//pip install -e .
 ``` 
 
 

environment-cpuonly.yaml

@@ -0,0 +1,33 @@
+name: ldm
+channels:
+  - pytorch
+  - defaults
+dependencies:
+  - python=3.8.5
+  - pip=20.3
+  - numpy=1.19.2
+  - pytorch
+  - torchvision
+  - torchaudio
+  - cpuonly
+  - -c pytorch
+  - pip:
+    - albumentations==0.4.3
+    - diffusers
+    - opencv-python==4.1.2.30
+    - pudb==2019.2
+    - invisible-watermark
+    - imageio==2.9.0
+    - imageio-ffmpeg==0.4.2
+    - pytorch-lightning==1.4.2
+    - 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
+    - torchmetrics==0.6.0
+    - 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 .