9 Commits

Author	SHA1	Message	Date
Aryan V S	038b42db94	Improve docs and type hints (#5759) ... * improvement: docs and type hints * improvement: docs and type hints minor refactor * improvement: docs and type hints * update with suggestions from review Co-Authored-By: Dhruv Nair <dhruv.nair@gmail.com> --------- Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>	2023-11-16 14:00:32 +05:30
0x1355	b95cbdf6fc	Set step_rules correctly for piecewise_constant scheduler (#3605) ... So that schedule_func() calls get_piecewise_constant_schedule() with correctly named kwarg.	2023-06-05 10:16:26 +05:30
Jason Kuan	a7b0671c07	add constant learning rate with custom rule (#3133) ... * add constant lr with rules * add constant with rules in TYPE_TO_SCHEDULER_FUNCTION * add constant lr rate with rule * hotfix code quality * fix doc style * change name constant_with_rules to piecewise constant	2023-04-28 16:29:56 +05:30
Felix Blanke	25d927aa51	Add last_epoch argument to optimization.get_scheduler (#2850) ... Add last_epoch arg to optimization.get_scheduler. Allows the specification of the index of the last epoch when resuming training.	2023-03-28 16:46:41 +01:00
Patrick von Platen	eadf0e2555	[Copyright] 2023 (#2524)	2023-03-01 10:31:00 +01:00
Emil Bogomolov	d87cc15977	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>	2022-12-20 01:41:37 +01:00
Anton Lozhkov	ed22b4fd07	Revive make quality (#203) ... * Revive Make utils * Add datasets for training too	2022-08-17 15:22:04 +02:00
anton-l	9e31c6a749	refactor GLIDE text2im pipeline, remove classifier_free_guidance	2022-06-21 14:07:58 +02:00
anton-l	71289ba06e	add lr schedule utils	2022-06-21 11:35:56 +02:00