VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-06 13:47:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add more dpmsolver++2m samplers to solve SDXL sampling artifacts

Browse Source
This commit is contained in:
rockerBOO
2025-01-23 16:52:36 -05:00
parent e89653975d
commit 9c1168a088
1 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
library/train_util.py
View File

@@ -5980,6 +5980,31 @@ 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
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":