From fd8d19a6e7ed6f0f3e8516788a3d6ddb4ef52aa8 Mon Sep 17 00:00:00 2001
From: Jiarui Fang <fangjiarui123@gmail.com>
Date: Sun, 29 Jan 2023 13:52:15 +0800
Subject: [PATCH] [example] update lightning dependency for stable diffusion
 (#2522)

---
 examples/images/diffusion/docker/Dockerfile | 7 -------
 examples/images/diffusion/requirements.txt  | 1 +
 2 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/examples/images/diffusion/docker/Dockerfile b/examples/images/diffusion/docker/Dockerfile
index 17cc8bc8b..e8e6957b7 100644
--- a/examples/images/diffusion/docker/Dockerfile
+++ b/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
 
diff --git a/examples/images/diffusion/requirements.txt b/examples/images/diffusion/requirements.txt
index 60c4b903e..1a9233d57 100644
--- a/examples/images/diffusion/requirements.txt
+++ b/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 .