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

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


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

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

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

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