Add lms and dpmpp2_s karras scheduler

2026-04-23 03:00:31 -04:00 · 2023-06-17 21:00:16 +03:00
parent 4cbc802e36
commit 6b7cf3f3be
7 changed files with 13 additions and 5 deletions
--- a/invokeai/backend/install/legacy_arg_parsing.py
+++ b/invokeai/backend/install/legacy_arg_parsing.py
@@ -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",