feat: make additional schedulers available in UI

api/onnx_web/serve.py

@@ -3,8 +3,13 @@ from diffusers import (
     DDIMScheduler,
     DDPMScheduler,
     DPMSolverMultistepScheduler,
+    DPMSolverSinglestepScheduler,
     EulerDiscreteScheduler,
     EulerAncestralDiscreteScheduler,
+    HeunDiscreteScheduler,
+    KDPM2AncestralDiscreteScheduler,
+    KDPM2DiscreteScheduler,
+    KarrasVeScheduler,
     LMSDiscreteScheduler,
     PNDMScheduler,
     # onnx
@@ -54,8 +59,13 @@ pipeline_schedulers = {
     'ddim': DDIMScheduler,
     'ddpm': DDPMScheduler,
     'dpm-multi': DPMSolverMultistepScheduler,
+    'dpm-single': DPMSolverSinglestepScheduler,
     'euler': EulerDiscreteScheduler,
     'euler-a': EulerAncestralDiscreteScheduler,
+    'heun': HeunDiscreteScheduler,
+    'k-dpm-2-a': KDPM2AncestralDiscreteScheduler,
+    'k-dpm-2': KDPM2DiscreteScheduler,
+    'karras-ve': KarrasVeScheduler,
     'lms-discrete': LMSDiscreteScheduler,
     'pndm': PNDMScheduler,
 }

gui/src/strings.ts

@@ -12,8 +12,13 @@ export const SCHEDULER_LABELS: Record<string, string> = {
   'ddim': 'DDIM',
   'ddpm': 'DDPM',
   'dpm-multi': 'DPM Multistep',
+  'dpm-single': 'DPM Singlestep',
   'euler': 'Euler',
   'euler-a': 'Euler Ancestral',
-  'lms-discrete': 'LMS Discrete',
+  'heun': 'Heun',
+  'k-dpm-2-a': 'KDPM2 Ancestral',
+  'k-dpm-2': 'KDPM2',
+  'karras-ve': 'Karras Ve',
+  'lms-discrete': 'LMS',
   'pndm': 'PNDM',
 };