Gated Diffusion accelerate config accelerate launch --mixed_precision="fp16" --multi_gpu train.py --name <RUN_NAME> --config configs/train.yaml