github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[diffusers] fix ci and docker (#3085)

pull/3089/head
Fazzie-Maqianli 2023-03-10 10:35:15 +08:00 committed by GitHub
parent 3213347b49
commit 5d5f475d75
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 2 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
examples/images/diffusion/README.md
View File

@ -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

2
examples/images/diffusion/docker/Dockerfile
View File

@ -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

2
examples/images/diffusion/test_ci.sh
View File

@ -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