Support EDM-style training in DreamBooth LoRA SDXL script (#7126)
* add: dreambooth lora script for Playground v2.5
* fix: kwarg
* address suraj's comments.
* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* apply suraj's suggestion
* incorporate changes in the canonical script./
* tracker naming
* fix: schedule determination
* add: two simple tests
* remove playground script
* note about edm-style training
* address pedro's comments.
* address part of Suraj's comments.
* Apply suggestions from code review
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* remove guidance_scale.
* use mse_loss.
* add comments for preconditioning.
* quality
* Update examples/dreambooth/train_dreambooth_lora_sdxl.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* tackle v-pred.
* Empty-Commit
* support edm for sdxl too.
* address suraj's comments.
* Empty-Commit
---------
Co-authored-by: Suraj Patil <surajp815@gmail.com>