ColossalAI/examples/images/diffusion/train_ddp.sh

6 lines
129 B
Bash

HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1
python main.py --logdir /tmp -t -b /configs/train_ddp.yaml