Skip to content

GitLab

Unverified Commit fd8d19a6 authored by Jiarui Fang's avatar Jiarui Fang Committed by GitHub
Browse files

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

parent aa0f6686
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 7 deletions
examples/images/diffusion/docker/Dockerfile
View file @ fd8d19a6
...@@ -18,13 +18,6 @@ RUN git clone https://github.com/NVIDIA/apex && \ ...@@ -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 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 # install titans
RUN pip install --no-cache-dir titans RUN pip install --no-cache-dir titans
......
examples/images/diffusion/requirements.txt
View file @ fd8d19a6
...@@ -13,6 +13,7 @@ transformers==4.19.2 ...@@ -13,6 +13,7 @@ transformers==4.19.2
webdataset==0.2.5 webdataset==0.2.5
open-clip-torch==2.7.0 open-clip-torch==2.7.0
gradio==3.11 gradio==3.11
lightning==1.9.0
datasets datasets
colossalai colossalai
-e . -e .
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment