"ci/install_dependencies_python2.sh" did not exist on "6a510ffc93e890008dab58eb7d722bd43997a956"
__init__.py 1.28 KB
Newer Older
Patrick von Platen's avatar
Patrick von Platen committed
1
2
3
# flake8: noqa
# There's no way to ignore "F401 '...' imported but unused" warnings in this
# module, but to preserve other warnings. So, don't check this module at all.
4
from .utils import is_inflect_available, is_scipy_available, is_transformers_available, is_unidecode_available
Patrick von Platen's avatar
Patrick von Platen committed
5
6


anton-l's avatar
anton-l committed
7
__version__ = "0.1.3"
Patrick von Platen's avatar
Patrick von Platen committed
8

Patrick von Platen's avatar
Patrick von Platen committed
9
from .modeling_utils import ModelMixin
Patrick von Platen's avatar
Patrick von Platen committed
10
from .models import AutoencoderKL, UNet2DConditionModel, UNet2DModel, VQModel
Omar Sanseviero's avatar
Omar Sanseviero committed
11
12
13
14
15
from .optimization import (
    get_constant_schedule,
    get_constant_schedule_with_warmup,
    get_cosine_schedule_with_warmup,
    get_cosine_with_hard_restarts_schedule_with_warmup,
16
    get_linear_schedule_with_warmup,
Omar Sanseviero's avatar
Omar Sanseviero committed
17
18
19
    get_polynomial_decay_schedule_with_warmup,
    get_scheduler,
)
Patrick von Platen's avatar
Patrick von Platen committed
20
from .pipeline_utils import DiffusionPipeline
21
22
23
24
25
26
27
28
29
from .pipelines import DDIMPipeline, DDPMPipeline, KarrasVePipeline, LDMPipeline, PNDMPipeline, ScoreSdeVePipeline
from .schedulers import (
    DDIMScheduler,
    DDPMScheduler,
    KarrasVeScheduler,
    PNDMScheduler,
    SchedulerMixin,
    ScoreSdeVeScheduler,
)
30
31
32
33
34


if is_scipy_available():
    from .schedulers import LMSDiscreteScheduler

Omar Sanseviero's avatar
Omar Sanseviero committed
35
from .training_utils import EMAModel
Patrick von Platen's avatar
Patrick von Platen committed
36

Patrick von Platen's avatar
Patrick von Platen committed
37

Patrick von Platen's avatar
Patrick von Platen committed
38
if is_transformers_available():
Suraj Patil's avatar
Suraj Patil committed
39
40
    from .pipelines import LDMTextToImagePipeline, StableDiffusionPipeline

41

Patrick von Platen's avatar
Patrick von Platen committed
42
43
else:
    from .utils.dummy_transformers_objects import *