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

Merge ca6b68ef7d into fa53f71ec0

Browse Source
This commit is contained in:
Dave Lage
2026-04-05 01:13:19 +00:00
committed by GitHub
parent fa53f71ec0 ca6b68ef7d
commit 6c5c307f94
1 changed files with 26 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
library/train_util.py
View File

@@ -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":