examples/custom_diffusion/train_custom_diffusion.py · 40c13fe5b41872d99dde75bf32f5c31626cf0043 · renzhc / diffusers_dcu

[train_custom_diffusion.py] Fix the LR schedulers when `num_train_epochs` is... · 40c13fe5

Anand Kumar authored Aug 29, 2024

[train_custom_diffusion.py] Fix the LR schedulers when `num_train_epochs` is passed in a distributed training env (#9308)

* Update train_custom_diffusion.py to fix the LR schedulers for `num_train_epochs`

* Fix saving text embeddings during safe serialization

* Fixed formatting

40c13fe5

train_custom_diffusion.py 59.3 KB

Replace train_custom_diffusion.py