mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
489 B
16 lines
489 B
#!/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
|