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

pull/2523/head
Jiarui Fang 2023-01-29 13:52:15 +08:00 committed by GitHub
parent aa0f6686f9
commit fd8d19a6e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 7 deletions

View File

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

View File

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