"""The MOSS Turbo loop is deliberately separate from ComfyUI's Euler sampler."""


def moss_turbo_sample(noise, mixing, codes, predict, tensor_ops, callback=None):
    if tuple(mixing.shape) != (128, 128) or len(codes) != 8:
        raise ValueError("MOSS Turboには128×128のQと8個のstep_codesが必要です。")
    if noise.shape[1] != 128:
        raise ValueError("MOSS Turboのlatent channelsは128です。")
    clean = tensor_ops.zeros_like(noise)
    for index in range(8):
        mixed = noise + tensor_ops.einsum("oc,bchw->bohw", mixing, clean)
        # Identical initial noise at every step, no Euler dt and no CFG branch.
        velocity = predict(mixed, codes[index])
        clean = noise - velocity.float() if hasattr(velocity, "float") else noise - velocity
        if callback is not None:
            callback(index + 1, 8)
    return clean