Add DDPM Scheduler

invokeai/backend/args.py

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

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, DEISMultistepScheduler
+    DPMSolverSinglestepScheduler, DEISMultistepScheduler, DDPMScheduler
 
 SCHEDULER_MAP = dict(
     ddim=(DDIMScheduler, dict()),
+    ddpm=(DDPMScheduler, dict()),
     deis=(DEISMultistepScheduler, dict()),
     lms=(LMSDiscreteScheduler, dict()),
     pndm=(PNDMScheduler, dict()),

invokeai/backend/web/modules/parameters.py

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

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

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

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

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