/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
examples/images/diffusion/test_ci.sh
15 строк
489 B
Hongxin Liu
[devops] fix extention building (#5427)
05 мар 2024, 10:35
Не верифицирован
05 мар 2024, 10:35
070df68
Код
Авторство
О чём код?
#!/bin/bash set -euxo pipefail conda env create -f environment.yaml conda activate ldm conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch pip install transformers diffusers invisible-watermark BUILD_EXT=1 pip install colossalai wget https://huggingface.co/stabilityai/stable-diffusion-2-base/resolve/main/512-base-ema.ckpt python main.py --logdir /tmp --train --base configs/Teyvat/train_colossalai_teyvat.yaml --ckpt 512-base-ema.ckpt