Fix dpmsolver++_2m_lu

2026-04-15 08:36:41 +00:00 · 2025-01-23 17:12:53 -05:00
parent 9c1168a088
commit ca6b68ef7d
1 changed files with 1 additions and 0 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -5984,6 +5984,7 @@ def get_my_scheduler(
        scheduler_cls = DPMSolverMultistepScheduler
    elif sample_sampler == "dpmsolver++_2m_lu":
        scheduler_cls = DPMSolverMultistepScheduler
+        sched_init_args["use_lu_lambdas"] = True
    elif sample_sampler == "dpmsolver++_2m_k":
        scheduler_cls = DPMSolverMultistepScheduler
        sched_init_args["use_karras_sigmas"] = True