__init__.py 425 Bytes
Newer Older
1
2
from typing import TYPE_CHECKING

Dhruv Nair's avatar
Dhruv Nair committed
3
from ...utils import (
4
    DIFFUSERS_SLOW_IMPORT,
Dhruv Nair's avatar
Dhruv Nair committed
5
6
7
8
    _LazyModule,
)


9
_import_structure = {"pipeline_ddpm": ["DDPMPipeline"]}
Dhruv Nair's avatar
Dhruv Nair committed
10

11
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
12
    from .pipeline_ddpm import DDPMPipeline
Dhruv Nair's avatar
Dhruv Nair committed
13

14
15
else:
    import sys
Dhruv Nair's avatar
Dhruv Nair committed
16

17
18
19
20
21
22
    sys.modules[__name__] = _LazyModule(
        __name__,
        globals()["__file__"],
        _import_structure,
        module_spec=__spec__,
    )