apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat: add support for UniPC multistep scheduler (#226)

This commit is contained in:
Sean Sube 2023-03-09 08:00:59 -06:00
parent 3d0f2434af
commit fe437e8cd4
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/diffusers/load.py
View File

@ -30,6 +30,11 @@ try:
except ImportError: except ImportError:
from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
try:
from diffusers import UniPCMultistepScheduler
except ImportError:
from ..diffusers.stub_scheduler import StubScheduler as UniPCMultistepScheduler
from ..params import DeviceParams, Size from ..params import DeviceParams, Size
from ..server import ServerContext from ..server import ServerContext
from ..utils import run_gc from ..utils import run_gc
@ -54,6 +59,7 @@ pipeline_schedulers = {
"karras-ve": KarrasVeScheduler, "karras-ve": KarrasVeScheduler,
"lms-discrete": LMSDiscreteScheduler, "lms-discrete": LMSDiscreteScheduler,
"pndm": PNDMScheduler, "pndm": PNDMScheduler,
"unipc-multi": UniPCMultistepScheduler,
} }

1
gui/src/strings/en.ts
View File

@ -198,6 +198,7 @@ export const I18N_STRINGS_EN = {
'ipndm': 'iPNDM', 'ipndm': 'iPNDM',
'lms-discrete': 'LMS', 'lms-discrete': 'LMS',
'pndm': 'PNDM', 'pndm': 'PNDM',
'unipc-multi': 'UniPC Multistep',
}, },
tab: { tab: {
blend: 'Blend', blend: 'Blend',