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

[example] polish diffusion readme

pull/1842/head
jiaruifang 2022-11-09 09:38:05 +08:00
parent cddb4b6f6f
commit 27211d6267
1 changed files with 21 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -39,18 +39,16 @@ pip install transformers==4.19.2 diffusers invisible-watermark
pip install -e .
```
### Install Colossal-AI
### Install [Colossal-AI v0.1.10](https://colossalai.org/download/) From Our Official Website
```
git clone https://github.com/hpcaitech/ColossalAI.git
git checkout v0.1.10
pip install .
pip install colossalai==0.1.10+torch1.11cu11.3 -f https://release.colossalai.org
```
### Install Colossal-AI [Lightning](https://github.com/Lightning-AI/lightning)
### Install [Lightning](https://github.com/Lightning-AI/lightning)
We use the Sep. 2022 version with commit id as `b04a7aa`.
```
git clone -b colossalai https://github.com/Fazziekey/lightning.git
pip install .
git clone https://github.com/Lightning-AI/lightning && cd lightning && git reset --hard b04a7aa
pip install -r requirements.txt && pip install .
```
## Dataset
@ -112,5 +110,3 @@ Thanks for open-sourcing!
year={2022}
}
```