Skip to content

Block Absorbing Process

discrete_diffusion.forward_process.block_absorbing

Block-absorbing forward process.

Minimal implementation: applies the absorbing process with a single shared time t across the entire sequence (i.e., one block == whole sequence). Returns (masked_ids, p_mask, t_out) where t_out is broadcast per-position.