/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
examples/images/diffusion/environment.yaml
32 строки
662 B
Camille Zhong
Fix the version of lightning and colossalai in Stable Diffusion environment requirement (#3073)
10 мар 2023, 04:55
Не верифицирован
10 мар 2023, 04:55
e58a3c8
Код
Авторство
О чём код?
name: ldm channels: - pytorch - defaults dependencies: - python=3.9.12 - pip=20.3 - cudatoolkit=11.3 - pytorch=1.12.1 - torchvision=0.13.1 - numpy=1.23.1 - pip: - albumentations==1.3.0 - opencv-python==4.6.0.66 - imageio==2.9.0 - imageio-ffmpeg==0.4.2 - omegaconf==2.1.1 - test-tube>=0.7.5 - streamlit==1.12.1 - einops==0.3.0 - transformers - webdataset==0.2.5 - kornia==0.6 - open_clip_torch==2.0.2 - invisible-watermark>=0.1.5 - streamlit-drawable-canvas==0.8.0 - torchmetrics==0.7.0 - prefetch_generator - datasets - colossalai==0.2.5 - lightning==1.9.0 - -e .