apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat: add iPNDM scheduler

This commit is contained in:
Sean Sube 2023-02-14 07:27:51 -06:00
parent 1289c560bd
commit f2d0025545
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/diffusion/load.py
View File

@ -11,6 +11,7 @@ from diffusers import (
EulerAncestralDiscreteScheduler, EulerAncestralDiscreteScheduler,
EulerDiscreteScheduler, EulerDiscreteScheduler,
HeunDiscreteScheduler, HeunDiscreteScheduler,
IPNDMScheduler,
KarrasVeScheduler, KarrasVeScheduler,
KDPM2AncestralDiscreteScheduler, KDPM2AncestralDiscreteScheduler,
KDPM2DiscreteScheduler, KDPM2DiscreteScheduler,
@ -40,6 +41,7 @@ pipeline_schedulers = {
"euler": EulerDiscreteScheduler, "euler": EulerDiscreteScheduler,
"euler-a": EulerAncestralDiscreteScheduler, "euler-a": EulerAncestralDiscreteScheduler,
"heun": HeunDiscreteScheduler, "heun": HeunDiscreteScheduler,
"ipndm": IPNDMScheduler,
"k-dpm-2-a": KDPM2AncestralDiscreteScheduler, "k-dpm-2-a": KDPM2AncestralDiscreteScheduler,
"k-dpm-2": KDPM2DiscreteScheduler, "k-dpm-2": KDPM2DiscreteScheduler,
"karras-ve": KarrasVeScheduler, "karras-ve": KarrasVeScheduler,

3
gui/src/strings.ts
View File

@ -15,6 +15,7 @@ export const MODEL_LABELS: Record<string, string> = {
'correction-codeformer': 'CodeFormer', 'correction-codeformer': 'CodeFormer',
'correction-gfpgan-v1-3': 'GFPGAN v1.3', 'correction-gfpgan-v1-3': 'GFPGAN v1.3',
// extras // extras
'diffusion-stablydiffused-aesthetic-v2-6': 'Aesthetic Mix v2.6',
'diffusion-anything': 'Anything', 'diffusion-anything': 'Anything',
'diffusion-anything-v3': 'Anything v3', 'diffusion-anything-v3': 'Anything v3',
'diffusion-anything-v4': 'Anything v4', 'diffusion-anything-v4': 'Anything v4',
@ -28,6 +29,7 @@ export const MODEL_LABELS: Record<string, string> = {
'diffusion-openjourney-v1': 'OpenJourney v1', 'diffusion-openjourney-v1': 'OpenJourney v1',
'diffusion-openjourney-v2': 'OpenJourney v2', 'diffusion-openjourney-v2': 'OpenJourney v2',
'diffusion-pastel-mix': 'Pastel Mix', 'diffusion-pastel-mix': 'Pastel Mix',
'diffusion-unstable-ink-dream-v6': 'Unstable Ink Dream v6',
}; };
export const PLATFORM_LABELS: Record<string, string> = { 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-a': 'KDPM2 Ancestral',
'k-dpm-2': 'KDPM2', 'k-dpm-2': 'KDPM2',
'karras-ve': 'Karras Ve', 'karras-ve': 'Karras Ve',
'ipndm': 'iPNDM',
'lms-discrete': 'LMS', 'lms-discrete': 'LMS',
'pndm': 'PNDM', 'pndm': 'PNDM',
}; };