Add DEIS Scheduler

invokeai/backend/args.py

@@ -108,6 +108,7 @@ APP_VERSION = invokeai.version.__version__
 
 SAMPLER_CHOICES = [
     "ddim",
+    "deis",
     "lms",
     "pndm",
     "heun",

invokeai/backend/stable_diffusion/schedulers/schedulers.py

@@ -1,10 +1,11 @@
 from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, KDPM2DiscreteScheduler, \
     KDPM2AncestralDiscreteScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler, \
     HeunDiscreteScheduler, LMSDiscreteScheduler, PNDMScheduler, UniPCMultistepScheduler, \
-    DPMSolverSinglestepScheduler
+    DPMSolverSinglestepScheduler, DEISMultistepScheduler
 
 SCHEDULER_MAP = dict(
     ddim=(DDIMScheduler, dict()),
+    deis=(DEISMultistepScheduler, dict()),
     lms=(LMSDiscreteScheduler, dict()),
     pndm=(PNDMScheduler, dict()),
     heun=(HeunDiscreteScheduler, dict()),

invokeai/backend/web/modules/parameters.py

@@ -4,6 +4,7 @@ from .parse_seed_weights import parse_seed_weights
 
 SAMPLER_CHOICES = [
     "ddim",
+    "deis",
     "lms",
     "pndm",
     "heun",

invokeai/frontend/web/src/app/constants.ts

@@ -2,6 +2,7 @@
 
 export const DIFFUSERS_SCHEDULERS: Array<string> = [
   'ddim',
+  'deis',
   'lms',
   'pndm',
   'heun',

invokeai/frontend/web/src/app/types/invokeai.ts

@@ -47,6 +47,7 @@ export type CommonGeneratedImageMetadata = {
   postprocessing: null | Array<ESRGANMetadata | FacetoolMetadata>;
   sampler:
     | 'ddim'
+    | 'deis'
     | 'lms'
     | 'pndm'
     | 'heun'