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_dreambooth.py \
  --snr_gamma=5.0  Prior preservation loss Prior preservation loss is a method that uses a model’s own generated samples to help it learn how to generate more diverse images. Because these generat