github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-18 09:21:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
1. Modify the README of stable diffusion
2. Fix the version of pytorch lightning&lightning and colossalai version to enable codes running successfully.
This commit is contained in:
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 ##### 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 #### 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 - torchmetrics==0.7.0
- prefetch_generator - prefetch_generator
- datasets - datasets
- colossalai - colossalai==0.2.5
- lightning==1.9.0
- -e . - -e .