Merge branch 'main' into logging-facelift

This commit is contained in:
blessedcoolant
2023-05-15 02:08:29 +12:00
committed by GitHub
25 changed files with 436 additions and 112 deletions

View File

@ -113,6 +113,7 @@ SAMPLER_CHOICES = [
"lms",
"pndm",
"heun",
"heun_k",
"euler",
"euler_k",
"euler_a",

View File

@ -9,7 +9,8 @@ SCHEDULER_MAP = dict(
deis=(DEISMultistepScheduler, dict()),
lms=(LMSDiscreteScheduler, dict()),
pndm=(PNDMScheduler, dict()),
heun=(HeunDiscreteScheduler, dict()),
heun=(HeunDiscreteScheduler, dict(use_karras_sigmas=False)),
heun_k=(HeunDiscreteScheduler, dict(use_karras_sigmas=True)),
euler=(EulerDiscreteScheduler, dict(use_karras_sigmas=False)),
euler_k=(EulerDiscreteScheduler, dict(use_karras_sigmas=True)),
euler_a=(EulerAncestralDiscreteScheduler, dict()),

View File

@ -9,6 +9,7 @@ SAMPLER_CHOICES = [
"lms",
"pndm",
"heun",
'heun_k',
"euler",
"euler_k",
"euler_a",