2023-04-15 19:32:22 +00:00
|
|
|
from diffusers import * # NOQA
|
|
|
|
|
|
|
|
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
|
|
|
|
2023-12-30 05:09:00 +00:00
|
|
|
try:
|
|
|
|
from diffusers.models.modeling_outputs import AutoencoderKLOutput
|
|
|
|
except ImportError:
|
|
|
|
from diffusers.models.autoencoder_kl import AutoencoderKLOutput
|
2023-04-15 19:32:22 +00:00
|
|
|
|
2023-12-30 05:09:00 +00:00
|
|
|
try:
|
|
|
|
from diffusers.models.autoencoders.vae import DecoderOutput
|
|
|
|
except ImportError:
|
|
|
|
from diffusers.models.vae import DecoderOutput
|
|
|
|
|
|
|
|
try:
|
2023-04-15 19:42:14 +00:00
|
|
|
from diffusers.models.attention_processor import AttnProcessor
|
2023-12-30 05:09:00 +00:00
|
|
|
except ImportError:
|
2023-04-15 19:42:14 +00:00
|
|
|
from diffusers.models.cross_attention import CrossAttnProcessor as AttnProcessor
|