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