[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_tr…
f7ebb2ad
Fix saving text embeddings during safe serialization
9e8d3442
sayakpaul
approved these changes
on 2024-08-29
Merge branch 'main' into AnandK27-custom-diffusion-LR-fix
691b693e
Fixed formatting
94e04e14
sayakpaul
merged
40c13fe5
into main 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub