[diffusers] fix ci and docker (#3085)

hpcaitech · Mar 10, 2023 · 5d5f475 · 5d5f475
1 parent 3213347
commit 5d5f475
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 21 deletions.
diff --git a/examples/images/diffusion/README.md b/examples/images/diffusion/README.md
@@ -55,24 +55,7 @@ conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit
 pip install transformers diffusers invisible-watermark
 ```
 
-#### Step 2: install lightning
-
-Install Lightning version later than 2022.01.04. We suggest you install lightning from source.
-
-##### From Source
-```
-git clone https://github.com/Lightning-AI/lightning.git
-pip install -r requirements.txt
-python setup.py install
-```
-
-##### From pip
-
-```
-pip install pytorch-lightning==2.0.0rc0
-```
-
-#### Step 3:Install [Colossal-AI](https://colossalai.org/download/) From Our Official Website
+#### Step 2:Install [Colossal-AI](https://colossalai.org/download/) From Our Official Website
 
 ##### From pip
 

diff --git a/examples/images/diffusion/docker/Dockerfile b/examples/images/diffusion/docker/Dockerfile
@@ -15,7 +15,7 @@ RUN git clone https://github.com/NVIDIA/apex && \
 #     && cd ./ColossalAI \
 #     && pip install -v --no-cache-dir .
 
-RUN pip install colossalai==0.1.12+torch1.12cu11.3 -f https://release.colossalai.org
+RUN pip install colossalai
 
 
 # install titans

diff --git a/examples/images/diffusion/test_ci.sh b/examples/images/diffusion/test_ci.sh
@@ -10,8 +10,6 @@ pip install transformers diffusers invisible-watermark
 
 CUDA_EXT=1  pip install colossalai
 
-pip install pytorch-lightning
-
 wget https://huggingface.co/stabilityai/stable-diffusion-2-base/resolve/main/512-base-ema.ckpt
 
 python main.py --logdir /tmp --train --base configs/Teyvat/train_colossalai_teyvat.yaml --ckpt 512-base-ema.ckpt