Spaces:
Running on Zero
Running on Zero
| 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 # noqa F403 | |
| _dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects)) | |
| else: | |
| _import_structure["modular_blocks_ltx2"] = [ | |
| "LTX2AutoBlocks", | |
| "LTX2Blocks", | |
| "LTX2ConditionBlocks", | |
| "LTX2ImageToVideoBlocks", | |
| "LTX2InContextBlocks", | |
| ] | |
| _import_structure["modular_blocks_ltx25"] = ["LTX25AutoBlocks"] | |
| _import_structure["modular_pipeline"] = ["LTX2ModularPipeline", "LTX25ModularPipeline"] | |
| 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 .modular_blocks_ltx2 import ( | |
| LTX2AutoBlocks, | |
| LTX2Blocks, | |
| LTX2ConditionBlocks, | |
| LTX2ImageToVideoBlocks, | |
| LTX2InContextBlocks, | |
| ) | |
| from .modular_blocks_ltx25 import LTX25AutoBlocks | |
| from .modular_pipeline import LTX2ModularPipeline, LTX25ModularPipeline | |
| 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) | |