feat: add support for UniPC multistep scheduler (#226)
This commit is contained in:
parent
3d0f2434af
commit
fe437e8cd4
|
@ -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,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -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',
|
||||||
|
|
Loading…
Reference in New Issue