| """Entry point for the Krea 2 reference-image edit modular blocks (loaded via `trust_remote_code`). | |
| Importing this module also registers `Krea2Transformer2DModel` into the `diffusers` namespace so that | |
| `ModularPipeline` component loading can resolve the transformer referenced as | |
| `["diffusers", "Krea2Transformer2DModel"]` in the base `krea/Krea-2-Turbo` `model_index.json`. | |
| """ | |
| from .transformer_krea2 import Krea2Transformer2DModel # noqa: F401 (registers into diffusers namespace) | |
| from .modular_blocks_krea2_edit import Krea2EditBlocks | |
| __all__ = ["Krea2EditBlocks", "Krea2Transformer2DModel"] | |