Fix dpmsolver++_2m_lu

kohya-ss · Jan 23, 2025 · ca6b68e · ca6b68e
1 parent 9c1168a
commit ca6b68e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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