github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-24 14:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/images/diffusion/docker/Dockerfile
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 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

1
examples/images/diffusion/requirements.txt
View File

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