apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat(api): add LCM scheduler

This commit is contained in:
Sean Sube 2023-11-21 23:14:15 -06:00
parent 54dc970a7e
commit fd97eebfa2
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -44,6 +44,7 @@ from .version_safe_diffusers import (
KarrasVeScheduler, KarrasVeScheduler,
KDPM2AncestralDiscreteScheduler, KDPM2AncestralDiscreteScheduler,
KDPM2DiscreteScheduler, KDPM2DiscreteScheduler,
LCMScheduler,
LMSDiscreteScheduler, LMSDiscreteScheduler,
OnnxRuntimeModel, OnnxRuntimeModel,
OnnxStableDiffusionImg2ImgPipeline, OnnxStableDiffusionImg2ImgPipeline,

5
api/onnx_web/diffusers/version_safe_diffusers.py
View File

@ -12,6 +12,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 LCMScheduler
except ImportError:
from ..diffusers.stub_scheduler import StubScheduler as LCMScheduler
try: try:
from diffusers import UniPCMultistepScheduler from diffusers import UniPCMultistepScheduler
except ImportError: except ImportError: