dir=${OUTPUT_DIR}

Training with multiple GPUs
	
accelerate allows for seamless multi-GPU training. Follow the instructions here
for running distributed training with accelerate. Here is an example command:


	Copied
export MODEL_NAME="CompVis/stable-diffusion-v1-4"
export dataset_name="lambdalabs/pokemon-blip-captions"

accelerate launch --mixed_precision="fp16" --multi_gpu  train_text_to_image.py \
  --pretrained_model_name_or_path=$MODEL_NAM