diff --git a/README.md b/README.md index 765d4aa..5a0fff2 100644 --- a/README.md +++ b/README.md @@ -25,6 +25,7 @@ A suitable [conda](https://conda.io/) environment named `ldm` can be created and activated with: ``` +conda install pytorch torchvision torchaudio cpuonly -c pytorch conda env create -f environment-cpuonly.yaml conda activate ldm ``` diff --git a/environment-cpuonly.yaml b/environment-cpuonly.yaml index cb83a1b..8275873 100644 --- a/environment-cpuonly.yaml +++ b/environment-cpuonly.yaml @@ -6,11 +6,6 @@ dependencies: - python=3.8.5 - pip=20.3 - numpy=1.19.2 - - pytorch - - torchvision - - torchaudio - - cpuonly - - -c pytorch - pip: - albumentations==0.4.3 - diffusers