1
0
Fork 0
onnx-web/api/onnx_web/diffusers/version_safe_diffusers.py

30 lines
835 B
Python
Raw Normal View History

import diffusers
from diffusers import * # NOQA
from packaging import version
is_diffusers_0_15 = version.parse(
version.parse(diffusers.__version__).base_version
) >= version.parse("0.15")
try:
2023-04-15 19:42:14 +00:00
from diffusers import DEISMultistepScheduler
except ImportError:
2023-04-15 19:42:14 +00:00
from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
2023-11-22 05:14:15 +00:00
try:
from diffusers import LCMScheduler
except ImportError:
from ..diffusers.stub_scheduler import StubScheduler as LCMScheduler
try:
2023-04-15 19:42:14 +00:00
from diffusers import UniPCMultistepScheduler
except ImportError:
2023-04-15 19:42:14 +00:00
from ..diffusers.stub_scheduler import StubScheduler as UniPCMultistepScheduler
if is_diffusers_0_15:
2023-04-15 19:42:14 +00:00
from diffusers.models.attention_processor import AttnProcessor
else:
2023-04-15 19:42:14 +00:00
from diffusers.models.cross_attention import CrossAttnProcessor as AttnProcessor