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):
10391039 use_ms : bool = True
10401040
10411041SCHEDULER_HANDLERS = {
1042- "normal" : SchedulerHandler (normal_scheduler ),
1042+ "simple" : SchedulerHandler (simple_scheduler ),
1043+ "sgm_uniform" : SchedulerHandler (partial (normal_scheduler , sgm = True )),
10431044 "karras" : SchedulerHandler (k_diffusion_sampling .get_sigmas_karras , use_ms = False ),
10441045 "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 ),
10471046 "ddim_uniform" : SchedulerHandler (ddim_scheduler ),
10481047 "beta" : SchedulerHandler (beta_scheduler ),
1048+ "normal" : SchedulerHandler (normal_scheduler ),
10491049 "linear_quadratic" : SchedulerHandler (linear_quadratic_schedule ),
10501050 "kl_optimal" : SchedulerHandler (kl_optimal_scheduler , use_ms = False ),
10511051}
You can’t perform that action at this time.
0 commit comments