File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -1039,13 +1039,13 @@ class SchedulerHandler(NamedTuple):
1039
1039
use_ms : bool = True
1040
1040
1041
1041
SCHEDULER_HANDLERS = {
1042
- "normal" : SchedulerHandler (normal_scheduler ),
1042
+ "simple" : SchedulerHandler (simple_scheduler ),
1043
+ "sgm_uniform" : SchedulerHandler (partial (normal_scheduler , sgm = True )),
1043
1044
"karras" : SchedulerHandler (k_diffusion_sampling .get_sigmas_karras , use_ms = False ),
1044
1045
"exponential" : SchedulerHandler (k_diffusion_sampling .get_sigmas_exponential , use_ms = False ),
1045
- "sgm_uniform" : SchedulerHandler (partial (normal_scheduler , sgm = True )),
1046
- "simple" : SchedulerHandler (simple_scheduler ),
1047
1046
"ddim_uniform" : SchedulerHandler (ddim_scheduler ),
1048
1047
"beta" : SchedulerHandler (beta_scheduler ),
1048
+ "normal" : SchedulerHandler (normal_scheduler ),
1049
1049
"linear_quadratic" : SchedulerHandler (linear_quadratic_schedule ),
1050
1050
"kl_optimal" : SchedulerHandler (kl_optimal_scheduler , use_ms = False ),
1051
1051
}
You can’t perform that action at this time.
0 commit comments