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

@@ -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

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 .