’re training on more than one GPU, add the --multi_gpu parameter to the accelerate launch command. To train on a local dataset, set the TRAIN_DIR and OUTPUT_DIR environment variables to the path of the dataset and where to save the model to.   Copied export MODEL_NAME="runwayml/stable-diffusion-v1-5"
export dataset_name="lambdalabs/pokemon-blip-captions"

accelerate launch --mixed_precision="fp16"  train_text_to_image.py \
  --pretrained_model_