From e58a3c804c4b107462d232c749a276cf7a704512 Mon Sep 17 00:00:00 2001
From: Camille Zhong <44392324+Camille7777@users.noreply.github.com>
Date: Fri, 10 Mar 2023 09:55:58 +0800
Subject: [PATCH] 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.
---
 examples/images/diffusion/README.md        | 2 +-
 examples/images/diffusion/environment.yaml | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/examples/images/diffusion/README.md b/examples/images/diffusion/README.md
index cc57f6d54..9e9900f2c 100644
--- a/examples/images/diffusion/README.md
+++ b/examples/images/diffusion/README.md
@@ -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
diff --git a/examples/images/diffusion/environment.yaml b/examples/images/diffusion/environment.yaml
index 5164be72e..d1ec69c1a 100644
--- a/examples/images/diffusion/environment.yaml
+++ b/examples/images/diffusion/environment.yaml
@@ -27,5 +27,6 @@ dependencies:
     - torchmetrics==0.7.0
     - prefetch_generator
     - datasets
-    - colossalai
+    - colossalai==0.2.5
+    - lightning==1.9.0
     - -e .