Add lms and dpmpp2_s karras scheduler

This commit is contained in:
Sergey Borisov
2023-06-17 21:00:16 +03:00
parent 4cbc802e36
commit 6b7cf3f3be
7 changed files with 13 additions and 5 deletions

View File

@ -9,6 +9,7 @@ SAMPLER_CHOICES = [
"ddpm",
"deis",
"lms",
"lms_k",
"pndm",
"heun",
"heun_k",
@ -18,6 +19,7 @@ SAMPLER_CHOICES = [
"kdpm_2",
"kdpm_2_a",
"dpmpp_2s",
"dpmpp_2s_k",
"dpmpp_2m",
"dpmpp_2m_k",
"unipc",