Unverified Commit e58a3c80 authored by Camille Zhong's avatar Camille Zhong Committed by GitHub
Browse files

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.
parent 8e4e8601
......@@ -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
......
......@@ -27,5 +27,6 @@ dependencies:
- torchmetrics==0.7.0
- prefetch_generator
- datasets
- colossalai
- colossalai==0.2.5
- lightning==1.9.0
- -e .
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment