diffusers
d87cc159 - expose polynomial:power and cosine_with_restarts:num_cycles params (#1737)

Commit
3 years ago
expose polynomial:power and cosine_with_restarts:num_cycles params (#1737) * expose polynomial:power and cosine_with_restarts:num_cycles using get_scheduler func, add it to train_dreambooth.py * fix formatting * fix style * Update src/diffusers/optimization.py Co-authored-by: Pedro Cuenca <pedro@huggingface.co> Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Author
Parents
Loading