From 4b957edfecbcfa1545c99d6252ecb5aeaa159d80 Mon Sep 17 00:00:00 2001
From: blessedcoolant <54517381+blessedcoolant@users.noreply.github.com>
Date: Fri, 12 May 2023 03:18:34 +1200
Subject: [PATCH] Add DDPM Scheduler

---
 invokeai/backend/args.py                                   | 1 +
 invokeai/backend/stable_diffusion/schedulers/schedulers.py | 3 ++-
 invokeai/backend/web/modules/parameters.py                 | 1 +
 invokeai/frontend/web/src/app/constants.ts                 | 1 +
 invokeai/frontend/web/src/app/types/invokeai.ts            | 1 +
 5 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/invokeai/backend/args.py b/invokeai/backend/args.py
index 4af9e43f42..856272c45b 100644
--- a/invokeai/backend/args.py
+++ b/invokeai/backend/args.py
@@ -108,6 +108,7 @@ APP_VERSION = invokeai.version.__version__
 
 SAMPLER_CHOICES = [
     "ddim",
+    "ddpm",
     "deis",
     "lms",
     "pndm",
diff --git a/invokeai/backend/stable_diffusion/schedulers/schedulers.py b/invokeai/backend/stable_diffusion/schedulers/schedulers.py
index 752e7cdb0c..fab28aca8c 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, DEISMultistepScheduler
+    DPMSolverSinglestepScheduler, DEISMultistepScheduler, DDPMScheduler
 
 SCHEDULER_MAP = dict(
     ddim=(DDIMScheduler, dict()),
+    ddpm=(DDPMScheduler, dict()),
     deis=(DEISMultistepScheduler, dict()),
     lms=(LMSDiscreteScheduler, dict()),
     pndm=(PNDMScheduler, dict()),
diff --git a/invokeai/backend/web/modules/parameters.py b/invokeai/backend/web/modules/parameters.py
index 940732898d..72211857a3 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",
+    "ddpm",
     "deis",
     "lms",
     "pndm",
diff --git a/invokeai/frontend/web/src/app/constants.ts b/invokeai/frontend/web/src/app/constants.ts
index f0dc48f6d5..a2dc04fd22 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<string> = [
   'ddim',
+  'ddpm',
   'deis',
   'lms',
   'pndm',
diff --git a/invokeai/frontend/web/src/app/types/invokeai.ts b/invokeai/frontend/web/src/app/types/invokeai.ts
index 2ae221b0a8..b79504d8f2 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<ESRGANMetadata | FacetoolMetadata>;
   sampler:
     | 'ddim'
+    | 'ddpm'
     | 'deis'
     | 'lms'
     | 'pndm'