Browse Source

[example] polish diffusion readme

pull/1842/head
jiaruifang 2 years ago
parent
commit
27211d6267
  1. 16
      examples/images/diffusion/README.md

16
examples/images/diffusion/README.md

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

Loading…
Cancel
Save