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

Fix the version of lightning and colossalai in Stable Diffusion environment requirement (#3073) 

1. Modify the README of stable diffusion
2. Fix the version of pytorch lightning&lightning and colossalai version to enable codes running successfully.
pull/3085/head
Camille Zhong 2023-03-10 09:55:58 +08:00 committed by GitHub
parent 8e4e8601b7
commit e58a3c804c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -69,7 +69,7 @@ python setup.py install
##### From pip
```
pip install pytorch-lightning
pip install pytorch-lightning==2.0.0rc0
```
#### Step 3:Install [Colossal-AI](https://colossalai.org/download/) From Our Official Website

3
examples/images/diffusion/environment.yaml
View File

@ -27,5 +27,6 @@ dependencies:
- torchmetrics==0.7.0
- prefetch_generator
- datasets
- colossalai
- colossalai==0.2.5
- lightning==1.9.0
- -e .