feat(api): add LCM scheduler
This commit is contained in:
parent
54dc970a7e
commit
fd97eebfa2
|
@ -44,6 +44,7 @@ from .version_safe_diffusers import (
|
|||
KarrasVeScheduler,
|
||||
KDPM2AncestralDiscreteScheduler,
|
||||
KDPM2DiscreteScheduler,
|
||||
LCMScheduler,
|
||||
LMSDiscreteScheduler,
|
||||
OnnxRuntimeModel,
|
||||
OnnxStableDiffusionImg2ImgPipeline,
|
||||
|
|
|
@ -12,6 +12,11 @@ try:
|
|||
except ImportError:
|
||||
from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
|
||||
|
||||
try:
|
||||
from diffusers import LCMScheduler
|
||||
except ImportError:
|
||||
from ..diffusers.stub_scheduler import StubScheduler as LCMScheduler
|
||||
|
||||
try:
|
||||
from diffusers import UniPCMultistepScheduler
|
||||
except ImportError:
|
||||
|
|
Loading…
Reference in New Issue