Skip to content

GitLab

"examples/t2i_adapter/requirements.txt" did not exist on "d0d3e24ec19daca42129afb89f1031d7e7c9995a"
Switch branch/tag
Find file BlameHistoryPermalink
  • Rafie Walker's avatar
    Implement SD3 loss weighting (#8528) · 6946facf
    Rafie Walker authored 
    

* Add lognorm and cosmap weighting

* Implement mode sampling

* Update examples/dreambooth/train_dreambooth_lora_sd3.py

* Update examples/dreambooth/train_dreambooth_lora_sd3.py

* Update examples/dreambooth/train_dreambooth_sd3.py

* Update examples/dreambooth/train_dreambooth_sd3.py

* Update examples/dreambooth/train_dreambooth_sd3.py

* Update examples/dreambooth/train_dreambooth_lora_sd3.py

* Update examples/dreambooth/train_dreambooth_sd3.py

* Update examples/dreambooth/train_dreambooth_sd3.py

* Update examples/dreambooth/train_dreambooth_lora_sd3.py

* keep timestamp sampling fully on cpu

---------
Co-authored-by: default avatarKashif Rasul <kashif.rasul@gmail.com>
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    6946facf
train_dreambooth_lora_sd3.py 65.1 KB