Merge ca6b68ef7d into fa53f71ec0

Fix dpmsolver++_2m_lu
Add more dpmsolver++2m samplers to solve SDXL sampling artifacts
2026-04-10 15:00:23 +00:00 · 2026-04-05 01:13:19 +00:00 · 2025-01-23 17:12:53 -05:00 · 2025-01-23 16:52:36 -05:00
1 changed files with 26 additions and 0 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -6292,6 +6292,32 @@ def get_my_scheduler(
    elif sample_sampler == "dpmsolver" or sample_sampler == "dpmsolver++":
        scheduler_cls = DPMSolverMultistepScheduler
        sched_init_args["algorithm_type"] = sample_sampler
+    elif sample_sampler == "dpmsolver++_2m":
+        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
+    elif sample_sampler == "dpmsolver++_2m_stable":
+        scheduler_cls = DPMSolverMultistepScheduler
+        sched_init_args["euler_at_final"] = True
+    elif sample_sampler == "dpmsolver++_2m_sde":
+        scheduler_cls = DPMSolverMultistepScheduler
+        sched_init_args["algorithm_type"] = "sde-dpmsolver++"
+    elif sample_sampler == "dpmsolver++_2m_sde_k":
+        scheduler_cls = DPMSolverMultistepScheduler
+        sched_init_args["algorithm_type"] = "sde-dpmsolver++"
+        sched_init_args["use_karras_sigmas"] = True
+    elif sample_sampler == "dpmsolver++_2m_sde_lu":
+        scheduler_cls = DPMSolverMultistepScheduler
+        sched_init_args["algorithm_type"] = "sde-dpmsolver++"
+        sched_init_args["use_lu_lambdas"] = True
+    elif sample_sampler == "dpmsolver++_2m_sde_stable":
+        scheduler_cls = DPMSolverMultistepScheduler
+        sched_init_args["algorithm_type"] = "sde-dpmsolver++"
+        sched_init_args["euler_at_final"] = True
    elif sample_sampler == "dpmsingle":
        scheduler_cls = DPMSolverSinglestepScheduler
    elif sample_sampler == "heun":
Author	SHA1	Message	Date
Dave Lage	6c5c307f94	Merge `ca6b68ef7d` into `fa53f71ec0`	2026-04-05 01:13:19 +00:00
rockerBOO	ca6b68ef7d	Fix dpmsolver++_2m_lu	2025-01-23 17:12:53 -05:00
rockerBOO	9c1168a088	Add more dpmsolver++2m samplers to solve SDXL sampling artifacts	2025-01-23 16:52:36 -05:00