[example] update lightning dependency for stable diffusion (#2522)

examples/images/diffusion/docker/Dockerfile

@@ -18,13 +18,6 @@ RUN git clone https://github.com/NVIDIA/apex && \
 RUN pip install colossalai==0.1.12+torch1.12cu11.3 -f https://release.colossalai.org
 
 
-# install our lightning, it will be merged to Lightning official repo.
-RUN git clone https://github.com/1SAA/lightning.git && \
-    cd lightning && \
-    git checkout strategy/colossalai && \
-    export PACKAGE_NAME=pytorch && \
-    pip install --no-cache-dir .
-
 # install titans
 RUN pip install --no-cache-dir titans
 

examples/images/diffusion/requirements.txt

@@ -13,6 +13,7 @@ transformers==4.19.2
 webdataset==0.2.5
 open-clip-torch==2.7.0
 gradio==3.11
+lightning==1.9.0
 datasets
 colossalai
 -e .