mirror of https://github.com/hpcaitech/ColossalAI
[example] update lightning dependency for stable diffusion (#2522)
parent
aa0f6686f9
commit
fd8d19a6e7
|
@ -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
|
||||
|
||||
|
|
|
@ -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 .
|
||||
|
|
Loading…
Reference in New Issue