diffusers
74e43a4f - Resolve v_prediction issue for min-SNR gamma weighted loss function (#5096)

Comment changes are shownComment changes are hidden
Commit
1 year ago
Resolve v_prediction issue for min-SNR gamma weighted loss function (#5096) * Resolve v_prediction issue for min-SNR gamma weighted loss function * Combine MSE loss calculation of epsilon and velocity, with a note about the application of the epsilon code to sample prediction * style --------- Co-authored-by: bghira <bghira@users.github.com> Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
Author
Parents
  • examples/text_to_image
    • File
      train_text_to_image_sdxl.py