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_flux3"] = ["Flux3AutoBlocks"] | |
| _import_structure["modular_blocks_flux3_distilled"] = ["Flux3DistilledAutoBlocks"] | |
| _import_structure["modular_pipeline"] = ["Flux3DistilledModularPipeline", "Flux3ModularPipeline"] | |
| 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_flux3 import Flux3AutoBlocks | |
| from .modular_blocks_flux3_distilled import Flux3DistilledAutoBlocks | |
| from .modular_pipeline import Flux3DistilledModularPipeline, Flux3ModularPipeline | |
| 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) | |