github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-24 19:17:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

[diffusers] fix ci and docker (#3085)

Browse Source
This commit is contained in:
Fazzie-Maqianli
2023-03-10 10:35:15 +08:00
committed by GitHub
parent 3213347b49
commit 5d5f475d75
3 changed files with 2 additions and 21 deletions
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