2023-04-15 19:32:22 +00:00
|
|
|
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
|
2023-04-15 19:32:22 +00:00
|
|
|
except ImportError:
|
2023-04-15 19:42:14 +00:00
|
|
|
from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
|
2023-04-15 19:32:22 +00:00
|
|
|
|
2023-12-10 05:51:09 +00:00
|
|
|
try:
|
|
|
|
from diffusers import DPMSolverSDEScheduler
|
2023-12-10 05:51:43 +00:00
|
|
|
except ImportError:
|
2023-12-10 05:51:09 +00:00
|
|
|
from ..diffusers.stub_scheduler import StubScheduler as DPMSolverSDEScheduler
|
|
|
|
|
2023-11-22 05:14:15 +00:00
|
|
|
try:
|
|
|
|
from diffusers import LCMScheduler
|
|
|
|
except ImportError:
|
|
|
|
from ..diffusers.stub_scheduler import StubScheduler as LCMScheduler
|
|
|
|
|
2023-04-15 19:32:22 +00:00
|
|
|
try:
|
2023-04-15 19:42:14 +00:00
|
|
|
from diffusers import UniPCMultistepScheduler
|
2023-04-15 19:32:22 +00:00
|
|
|
except ImportError:
|
2023-04-15 19:42:14 +00:00
|
|
|
from ..diffusers.stub_scheduler import StubScheduler as UniPCMultistepScheduler
|
2023-04-15 19:32:22 +00:00
|
|
|
|
|
|
|
|
|
|
|
if is_diffusers_0_15:
|
2023-04-15 19:42:14 +00:00
|
|
|
from diffusers.models.attention_processor import AttnProcessor
|
2023-04-15 19:32:22 +00:00
|
|
|
else:
|
2023-04-15 19:42:14 +00:00
|
|
|
from diffusers.models.cross_attention import CrossAttnProcessor as AttnProcessor
|