"""Custom pipeline entrypoint for Diffusers dynamic loading."""

from .custom_pipeline.pipeline_nit import NiTPipeline
from .custom_pipeline.scheduling_flow_match_nit import NiTFlowMatchScheduler
from .custom_pipeline.transformer_nit import NiTTransformer2DModel

try:
    import diffusers

    setattr(diffusers, "NiTPipeline", NiTPipeline)
    setattr(diffusers, "NiTTransformer2DModel", NiTTransformer2DModel)
    setattr(diffusers, "NiTFlowMatchScheduler", NiTFlowMatchScheduler)
except Exception:
    pass


__all__ = [
    "NiTPipeline",
    "NiTTransformer2DModel",
    "NiTFlowMatchScheduler",
]