Files
ColossalAI/examples/images/diffusion/train_colossalai.sh
natalie_cao de84c0311a Polish Code
2023-04-12 18:19:46 +08:00

7 lines
198 B
Bash
Executable File

HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1
python main.py --logdir /tmp --train --base configs/Teyvat/train_colossalai_teyvat.yaml --ckpt diffuser_root_dir/512-base-ema.ckpt