From ca6b68ef7da70ea410145991c1b098590d46f0a2 Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Thu, 23 Jan 2025 17:12:53 -0500 Subject: [PATCH] Fix dpmsolver++_2m_lu --- library/train_util.py | 1 + 1 file changed, 1 insertion(+) diff --git a/library/train_util.py b/library/train_util.py index eeace35a..ac108988 100644 --- 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