add constant with rules in TYPE_TO_SCHEDULER_FUNCTION

jason9075 · jason9075 · commit 8dde211806c4 · 2023-04-17T20:30:35.000+08:00
diff --git a/src/diffusers/optimization.py b/src/diffusers/optimization.py
@@ -276,6 +276,7 @@ def lr_lambda(current_step: int):
     SchedulerType.POLYNOMIAL: get_polynomial_decay_schedule_with_warmup,
     SchedulerType.CONSTANT: get_constant_schedule,
     SchedulerType.CONSTANT_WITH_WARMUP: get_constant_schedule_with_warmup,
+    SchedulerType.CONSTANT_WITH_RULES: get_constant_schedule_with_rules,
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -276,6 +276,7 @@ def lr_lambda(current_step: int):`
`276`	`276`	`SchedulerType.POLYNOMIAL: get_polynomial_decay_schedule_with_warmup,`
`277`	`277`	`SchedulerType.CONSTANT: get_constant_schedule,`
`278`	`278`	`SchedulerType.CONSTANT_WITH_WARMUP: get_constant_schedule_with_warmup,`
	`279`	`+ SchedulerType.CONSTANT_WITH_RULES: get_constant_schedule_with_rules,`
`279`	`280`	`}`
`280`	`281`
`281`	`282`