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

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

AnandK27
AnandK27 Update train_custom_diffusion.py to fix the LR schedulers for `num_tr…
f7ebb2ad
AnandK27 Fix saving text embeddings during safe serialization
9e8d3442
sayakpaul
sayakpaul approved these changes on 2024-08-29
HuggingFaceDocBuilderDev
AnandK27 Merge branch 'main' into AnandK27-custom-diffusion-LR-fix
691b693e
AnandK27 Fixed formatting
94e04e14
AnandK27
AnandK27
sayakpaul sayakpaul merged 40c13fe5 into main 1 year ago
sayakpaul

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone