"...git@developer.sourcefind.cn:OpenDAS/autoawq_kernels.git" did not exist on "cb74e4eec6bd1446cc84d3d343eef96720a42d0d"
__init__.py 1.64 KB
Newer Older
1
2
3
4
5
6
7
from .utils import (
    is_inflect_available,
    is_onnx_available,
    is_scipy_available,
    is_transformers_available,
    is_unidecode_available,
)
Patrick von Platen's avatar
Patrick von Platen committed
8
9


anton-l's avatar
anton-l committed
10
__version__ = "0.4.0.dev0"
Patrick von Platen's avatar
Patrick von Platen committed
11

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


if is_scipy_available():
    from .schedulers import LMSDiscreteScheduler
40
else:
41
    from .utils.dummy_scipy_objects import *  # noqa F403
42

Omar Sanseviero's avatar
Omar Sanseviero committed
43
from .training_utils import EMAModel
Patrick von Platen's avatar
Patrick von Platen committed
44

Patrick von Platen's avatar
Patrick von Platen committed
45

Patrick von Platen's avatar
Patrick von Platen committed
46
if is_transformers_available():
47
48
49
50
51
52
    from .pipelines import (
        LDMTextToImagePipeline,
        StableDiffusionImg2ImgPipeline,
        StableDiffusionInpaintPipeline,
        StableDiffusionPipeline,
    )
Patrick von Platen's avatar
Patrick von Platen committed
53
else:
54
    from .utils.dummy_transformers_objects import *  # noqa F403
55
56
57
58
59
60


if is_transformers_available() and is_onnx_available():
    from .pipelines import StableDiffusionOnnxPipeline
else:
    from .utils.dummy_transformers_and_onnx_objects import *  # noqa F403