FilesExpand file tree

main

/

__init__.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

48 lines (40 loc) · 1.43 KB

main

/

__init__.py

Copy path

Top

File metadata and controls

• Code
• Blame

48 lines (40 loc) · 1.43 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

# 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.

from .utils import is_inflect_available, is_scipy_available, is_transformers_available, is_unidecode_available

__version__ = "0.2.4"

from .modeling_utils import ModelMixin

from .models import AutoencoderKL, UNet2DConditionModel, UNet2DModel, VQModel

from .optimization import (

get_constant_schedule,

get_constant_schedule_with_warmup,

get_cosine_schedule_with_warmup,

get_cosine_with_hard_restarts_schedule_with_warmup,

get_linear_schedule_with_warmup,

get_polynomial_decay_schedule_with_warmup,

get_scheduler,

)

from .pipeline_utils import DiffusionPipeline

from .pipelines import DDIMPipeline, DDPMPipeline, KarrasVePipeline, LDMPipeline, PNDMPipeline, ScoreSdeVePipeline

from .schedulers import (

DDIMScheduler,

DDPMScheduler,

KarrasVeScheduler,

PNDMScheduler,

SchedulerMixin,

ScoreSdeVeScheduler,

)

if is_scipy_available():

from .schedulers import LMSDiscreteScheduler

else:

from .utils.dummy_scipy_objects import *

from .training_utils import EMAModel

if is_transformers_available():

from .pipelines import (

LDMTextToImagePipeline,

StableDiffusionImg2ImgPipeline,

StableDiffusionInpaintPipeline,

StableDiffusionPipeline,

)

else:

from .utils.dummy_transformers_objects import *