diff --git a/invokeai/backend/args.py b/invokeai/backend/args.py index 0925e19edf..4af9e43f42 100644 --- a/invokeai/backend/args.py +++ b/invokeai/backend/args.py @@ -108,6 +108,7 @@ APP_VERSION = invokeai.version.__version__ SAMPLER_CHOICES = [ "ddim", + "deis", "lms", "pndm", "heun", diff --git a/invokeai/backend/stable_diffusion/schedulers/schedulers.py b/invokeai/backend/stable_diffusion/schedulers/schedulers.py index 5df6d045c1..752e7cdb0c 100644 --- a/invokeai/backend/stable_diffusion/schedulers/schedulers.py +++ b/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()), diff --git a/invokeai/backend/web/modules/parameters.py b/invokeai/backend/web/modules/parameters.py index 22bc2f353e..940732898d 100644 --- a/invokeai/backend/web/modules/parameters.py +++ b/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", diff --git a/invokeai/frontend/web/src/app/constants.ts b/invokeai/frontend/web/src/app/constants.ts index ab1c313abe..f0dc48f6d5 100644 --- a/invokeai/frontend/web/src/app/constants.ts +++ b/invokeai/frontend/web/src/app/constants.ts @@ -2,6 +2,7 @@ export const DIFFUSERS_SCHEDULERS: Array = [ 'ddim', + 'deis', 'lms', 'pndm', 'heun', diff --git a/invokeai/frontend/web/src/app/types/invokeai.ts b/invokeai/frontend/web/src/app/types/invokeai.ts index 998bb00d47..2ae221b0a8 100644 --- a/invokeai/frontend/web/src/app/types/invokeai.ts +++ b/invokeai/frontend/web/src/app/types/invokeai.ts @@ -47,6 +47,7 @@ export type CommonGeneratedImageMetadata = { postprocessing: null | Array; sampler: | 'ddim' + | 'deis' | 'lms' | 'pndm' | 'heun'