on, but Min-SNR is compatible with both prediction types. This weighting strategy is only supported by PyTorch and is unavailable in the Flax training script. Add the --snr_gamma parameter and set it to the recommended value of 5.0:   Copied accelerate launch train_text_to_image_sdxl.py \
  --snr_gamma=5.0  Training script The training script is also similar to the Text-to-image training guide, but it’s been modified to support SDXL training. T