ColossalAI/examples/images/diffusion/train_colossalai.sh

6 lines
197 B
Bash
Raw Normal View History

2023-01-31 02:00:37 +00:00
HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1
2023-02-03 07:34:54 +00:00
python main.py --logdir /tmp --train --base configs/Teyvat/train_colossalai_teyvat.yaml --ckpt diffuser_root_dir/512-base-ema.ckpt