/
ai-forever
/
diffusers-new
Обзор
Документация
Войти
/
ai-forever
/
diffusers-new
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
main
src/diffusers/pipelines/stable_cascade/__init__.py
50 строк
2 KB
Kashif Rasul
[Pipiline] Wuerstchen v3 aka Stable Cascasde pipeline (#6487)
06 мар 2024, 12:37
Не верифицирован
06 мар 2024, 12:37
40aa47b
Код
Авторство
О чём код?
from typing import TYPE_CHECKING from ...utils import ( DIFFUSERS_SLOW_IMPORT, OptionalDependencyNotAvailable, _LazyModule, get_objects_from_module, is_torch_available, is_transformers_available, ) _dummy_objects = {} _import_structure = {} try: if not (is_transformers_available() and is_torch_available()): raise OptionalDependencyNotAvailable() except OptionalDependencyNotAvailable: from ...utils import dummy_torch_and_transformers_objects _dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects)) else: _import_structure["pipeline_stable_cascade"] = ["StableCascadeDecoderPipeline"] _import_structure["pipeline_stable_cascade_combined"] = ["StableCascadeCombinedPipeline"] _import_structure["pipeline_stable_cascade_prior"] = ["StableCascadePriorPipeline"] if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: try: if not (is_transformers_available() and is_torch_available()): raise OptionalDependencyNotAvailable() except OptionalDependencyNotAvailable: from ...utils.dummy_torch_and_transformers_objects import * # noqa F403 else: from .pipeline_stable_cascade import StableCascadeDecoderPipeline from .pipeline_stable_cascade_combined import StableCascadeCombinedPipeline from .pipeline_stable_cascade_prior import StableCascadePriorPipeline else: import sys sys.modules[__name__] = _LazyModule( __name__, globals()["__file__"], _import_structure, module_spec=__spec__, ) for name, value in _dummy_objects.items(): setattr(sys.modules[__name__], name, value)