"""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"]