### original: <https://github.com/magic-research/piecewise-rectified-flow>

from .scheduler_perflow import PeRFlowScheduler
from .utils_perflow import merge_delta_weights_into_unet