feat: add support for UniPC multistep scheduler (#226)
This commit is contained in:
parent
3d0f2434af
commit
fe437e8cd4
|
@ -30,6 +30,11 @@ try:
|
|||
except ImportError:
|
||||
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 ..server import ServerContext
|
||||
from ..utils import run_gc
|
||||
|
@ -54,6 +59,7 @@ pipeline_schedulers = {
|
|||
"karras-ve": KarrasVeScheduler,
|
||||
"lms-discrete": LMSDiscreteScheduler,
|
||||
"pndm": PNDMScheduler,
|
||||
"unipc-multi": UniPCMultistepScheduler,
|
||||
}
|
||||
|
||||
|
||||
|
|
|
@ -198,6 +198,7 @@ export const I18N_STRINGS_EN = {
|
|||
'ipndm': 'iPNDM',
|
||||
'lms-discrete': 'LMS',
|
||||
'pndm': 'PNDM',
|
||||
'unipc-multi': 'UniPC Multistep',
|
||||
},
|
||||
tab: {
|
||||
blend: 'Blend',
|
||||
|
|
Loading…
Reference in New Issue