"megatron/training/initialize.py" did not exist on "ac3db1595427ace88fd98a5560da6b75c4704a52"
  • Sayak Paul's avatar
    Support EDM-style training in DreamBooth LoRA SDXL script (#7126) · ccb93dca
    Sayak Paul authored
    
    
    * add: dreambooth lora script for Playground v2.5
    
    * fix: kwarg
    
    * address suraj's comments.
    
    * Apply suggestions from code review
    Co-authored-by: default avatarSuraj 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: default avatarSuraj 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: default avatarSuraj Patil <surajp815@gmail.com>
    
    * tackle v-pred.
    
    * Empty-Commit
    
    * support edm for sdxl too.
    
    * address suraj's comments.
    
    * Empty-Commit
    
    ---------
    Co-authored-by: default avatarSuraj Patil <surajp815@gmail.com>
    ccb93dca
test_dreambooth_lora_edm.py 4.02 KB