From f2d00255457bb1c847ca12538d0ee82f596a0c98 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Tue, 14 Feb 2023 07:27:51 -0600
Subject: [PATCH] feat: add iPNDM scheduler

---
 api/onnx_web/diffusion/load.py | 2 ++
 gui/src/strings.ts             | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py
index d0658dc4..8a1f25e3 100644
--- a/api/onnx_web/diffusion/load.py
+++ b/api/onnx_web/diffusion/load.py
@@ -11,6 +11,7 @@ from diffusers import (
     EulerAncestralDiscreteScheduler,
     EulerDiscreteScheduler,
     HeunDiscreteScheduler,
+    IPNDMScheduler,
     KarrasVeScheduler,
     KDPM2AncestralDiscreteScheduler,
     KDPM2DiscreteScheduler,
@@ -40,6 +41,7 @@ pipeline_schedulers = {
     "euler": EulerDiscreteScheduler,
     "euler-a": EulerAncestralDiscreteScheduler,
     "heun": HeunDiscreteScheduler,
+    "ipndm": IPNDMScheduler,
     "k-dpm-2-a": KDPM2AncestralDiscreteScheduler,
     "k-dpm-2": KDPM2DiscreteScheduler,
     "karras-ve": KarrasVeScheduler,
diff --git a/gui/src/strings.ts b/gui/src/strings.ts
index b16934ba..86c4171b 100644
--- a/gui/src/strings.ts
+++ b/gui/src/strings.ts
@@ -15,6 +15,7 @@ export const MODEL_LABELS: Record<string, string> = {
   'correction-codeformer': 'CodeFormer',
   'correction-gfpgan-v1-3': 'GFPGAN v1.3',
   // extras
+  'diffusion-stablydiffused-aesthetic-v2-6': 'Aesthetic Mix v2.6',
   'diffusion-anything': 'Anything',
   'diffusion-anything-v3': 'Anything v3',
   'diffusion-anything-v4': 'Anything v4',
@@ -28,6 +29,7 @@ export const MODEL_LABELS: Record<string, string> = {
   'diffusion-openjourney-v1': 'OpenJourney v1',
   'diffusion-openjourney-v2': 'OpenJourney v2',
   'diffusion-pastel-mix': 'Pastel Mix',
+  'diffusion-unstable-ink-dream-v6': 'Unstable Ink Dream v6',
 };
 
 export const PLATFORM_LABELS: Record<string, string> = {
@@ -53,6 +55,7 @@ export const SCHEDULER_LABELS: Record<string, string> = {
   'k-dpm-2-a': 'KDPM2 Ancestral',
   'k-dpm-2': 'KDPM2',
   'karras-ve': 'Karras Ve',
+  'ipndm': 'iPNDM',
   'lms-discrete': 'LMS',
   'pndm': 'PNDM',
 };