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 . 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 pip install colossalai==0.1.10+torch1.11cu11.3 -f https://release.colossalai.org
git checkout v0.1.10
pip install .
``` ```
### 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 git clone https://github.com/Lightning-AI/lightning && cd lightning && git reset --hard b04a7aa
pip install . pip install -r requirements.txt && pip install .
``` ```
## Dataset ## Dataset
@ -112,5 +110,3 @@ Thanks for open-sourcing!
year={2022} year={2022}
} }
``` ```

Loading…
Cancel
Save