scheduler/scheduling_flow_match_euler_discrete_time_shifting.py

# Copyright (C) 2026 Boogu Team.
# This repository is a fork by Boogu Team; modifications have been made.

"""
Compatibility entry point for the released Boogu base pipeline scheduler.

The inference code in this repository owns the canonical scheduler
implementation. Keeping another full copy here easily causes drift between the
released pipeline package and the active codebase. Diffusers only needs this
module to expose `FlowMatchEulerDiscreteScheduler` for `model_index.json`, so
this file intentionally re-exports the repository class.
"""

try:
    from boogu.schedulers.scheduling_flow_match_euler_discrete_time_shifting import (
        FlowMatchEulerDiscreteScheduler,
        FlowMatchEulerDiscreteSchedulerOutput,
    )
except ModuleNotFoundError as exc:
    if exc.name != "boogu":
        raise
    raise ModuleNotFoundError(
        "Failed to import the canonical Boogu scheduler implementation. "
        "Run inference from the Boogu-Image repository root "
        "or add that directory to PYTHONPATH before loading the pipeline."
    ) from exc


__all__ = [
    "FlowMatchEulerDiscreteScheduler",
    "FlowMatchEulerDiscreteSchedulerOutput",
]