Skip to content

Latest commit

 

History

History
25 lines (17 loc) · 1.11 KB

File metadata and controls

25 lines (17 loc) · 1.11 KB

BlockRefinementScheduler

The BlockRefinementScheduler manages block-wise iterative refinement for discrete token diffusion. At each step it commits the most confident tokens and optionally edits already-committed tokens when the model predicts a different token with high confidence.

This scheduler is used by [BlockRefinementPipeline] and [LLaDA2Pipeline].

BlockRefinementScheduler

[[autodoc]] BlockRefinementScheduler

BlockRefinementSchedulerOutput

[[autodoc]] schedulers.scheduling_block_refinement.BlockRefinementSchedulerOutput