huggingface · kashif · Dec 16, 2025 · Dec 16, 2025 · Dec 16, 2025 · Dec 16, 2025
diff --git a/docs/source/en/api/pipelines/block_refinement.md b/docs/source/en/api/pipelines/block_refinement.md
@@ -0,0 +1,63 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# Block Refinement
+
+`BlockRefinementPipeline` performs block-wise iterative refinement over a masked token template, sampling and
+committing tokens based on confidence.
+
+## Config defaults
+
+You can set default sampling parameters when creating the pipeline. Passing `None` for a parameter in `__call__`
+falls back to `pipe.config`.
+
+```py
+from diffusers import BlockRefinementPipeline
+
+pipe = BlockRefinementPipeline(
+    model=model,
+    tokenizer=tokenizer,
+    gen_length=256,
+    block_length=32,
+    steps=16,
+    temperature=0.8,
+    sampling_method="multinomial",
+)
+
+out = pipe(prompt="Explain gradient descent.")
+print(out.texts[0])
+```
+
+## Callbacks
+
+Callbacks run after each refinement step and can inspect or override the current tokens.
+
+```py
+def on_step_end(pipe, step, timestep, callback_kwargs):
+    cur_x = callback_kwargs["cur_x"]
+    # Inspect or modify `cur_x` here.
+    return {"cur_x": cur_x}
+
+out = pipe(
+    prompt="Write a short poem.",
+    callback_on_step_end=on_step_end,
+    callback_on_step_end_tensor_inputs=["cur_x"],
+)
+```
+
+## BlockRefinementPipeline
+[[autodoc]] BlockRefinementPipeline
+    - all
+    - __call__
+
+## BlockRefinementPipelineOutput
+[[autodoc]] pipelines.BlockRefinementPipelineOutput
diff --git a/docs/source/en/api/pipelines/block_token_diffusion.md b/docs/source/en/api/pipelines/block_token_diffusion.md
@@ -0,0 +1,23 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# Block Token Diffusion
+
+`BlockTokenDiffusionPipeline` performs token diffusion by iterating over fixed-size blocks of the sequence.
+
+## BlockTokenDiffusionPipeline
+[[autodoc]] BlockTokenDiffusionPipeline
+    - all
+    - __call__
+
+## BlockTokenDiffusionPipelineOutput
+[[autodoc]] pipelines.BlockTokenDiffusionPipelineOutput
diff --git a/docs/source/en/api/pipelines/hybrid_token_diffusion.md b/docs/source/en/api/pipelines/hybrid_token_diffusion.md
@@ -0,0 +1,23 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# Hybrid Token Diffusion
+
+`HybridTokenDiffusionPipeline` is an alias of `TokenDiffusionPipeline` for hybrid-transition schedulers.
+
+## HybridTokenDiffusionPipeline
+[[autodoc]] HybridTokenDiffusionPipeline
+    - all
+    - __call__
+
+## TokenDiffusionPipelineOutput
+[[autodoc]] pipelines.TokenDiffusionPipelineOutput
diff --git a/docs/source/en/api/pipelines/llada2.md b/docs/source/en/api/pipelines/llada2.md
@@ -0,0 +1,23 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# LLaDA2
+
+`LLaDA2Pipeline` adapts block refinement sampling for LLaDA2-style token diffusion models.
+
+## LLaDA2Pipeline
+[[autodoc]] LLaDA2Pipeline
+    - all
+    - __call__
+
+## LLaDA2PipelineOutput
+[[autodoc]] pipelines.LLaDA2PipelineOutput
diff --git a/docs/source/en/api/pipelines/overview.md b/docs/source/en/api/pipelines/overview.md
@@ -34,6 +34,8 @@ The table below lists all the pipelines currently available in 🤗 Diffusers an
 | [AudioLDM2](audioldm2) | text2audio |
 | [AuraFlow](aura_flow) | text2image |
 | [BLIP Diffusion](blip_diffusion) | text2image |
+| [Block Refinement](block_refinement) | text2text |
+| [Block Token Diffusion](block_token_diffusion) | text2text |
 | [Bria 3.2](bria_3_2) | text2image |
 | [CogVideoX](cogvideox) | text2video |
 | [Consistency Models](consistency_models) | unconditional image generation |
@@ -52,6 +54,7 @@ The table below lists all the pipelines currently available in 🤗 Diffusers an
 | [DiT](dit) | text2image |
 | [Flux](flux) | text2image |
 | [Hunyuan-DiT](hunyuandit) | text2image |
+| [Hybrid Token Diffusion](hybrid_token_diffusion) | text2text |
 | [I2VGen-XL](i2vgenxl) | image2video |
 | [InstructPix2Pix](pix2pix) | image editing |
 | [Kandinsky 2.1](kandinsky) | text2image, image2image, inpainting, interpolation |
@@ -62,6 +65,7 @@ The table below lists all the pipelines currently available in 🤗 Diffusers an
 | [Latent Diffusion](latent_diffusion) | text2image, super-resolution |
 | [Latte](latte) | text2image |
 | [LEDITS++](ledits_pp) | image editing |
+| [LLaDA2](llada2) | text2text |
 | [Lumina-T2X](lumina) | text2image |
 | [Marigold](marigold) | depth-estimation, normals-estimation, intrinsic-decomposition |
 | [MultiDiffusion](panorama) | text2image |
@@ -83,6 +87,7 @@ The table below lists all the pipelines currently available in 🤗 Diffusers an
 | [T2I-Adapter](stable_diffusion/adapter) | text2image |
 | [Text2Video](text_to_video) | text2video, video2video |
 | [Text2Video-Zero](text_to_video_zero) | text2video |
+| [Token Diffusion](token_diffusion) | text2text |
 | [unCLIP](unclip) | text2image, image variation |
 | [UniDiffuser](unidiffuser) | text2image, image2text, image variation, text variation, unconditional image generation, unconditional audio generation |
 | [Value-guided planning](value_guided_sampling) | value guided sampling |

diff --git a/docs/source/en/api/pipelines/token_diffusion.md b/docs/source/en/api/pipelines/token_diffusion.md
@@ -0,0 +1,24 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# Token Diffusion
+
+`TokenDiffusionPipeline` provides a generic token-space diffusion sampler for discrete denoising over token IDs. It
+pairs a token denoiser model with a token diffusion scheduler.
+
+## TokenDiffusionPipeline
+[[autodoc]] TokenDiffusionPipeline
+    - all
+    - __call__
+
+## TokenDiffusionPipelineOutput
+[[autodoc]] pipelines.TokenDiffusionPipelineOutput
diff --git a/docs/source/en/api/schedulers/block_token_diffusion.md b/docs/source/en/api/schedulers/block_token_diffusion.md
@@ -0,0 +1,21 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# BlockTokenDiffusionScheduler
+
+`BlockTokenDiffusionScheduler` extends `TokenDiffusionScheduler` with block-wise updates over token positions.
+
+## BlockTokenDiffusionScheduler
+[[autodoc]] BlockTokenDiffusionScheduler
+
+## TokenDiffusionSchedulerOutput
+[[autodoc]] schedulers.scheduling_token_diffusion.TokenDiffusionSchedulerOutput
diff --git a/docs/source/en/api/schedulers/hybrid_token_diffusion.md b/docs/source/en/api/schedulers/hybrid_token_diffusion.md
@@ -0,0 +1,22 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# HybridTokenDiffusionScheduler
+
+`HybridTokenDiffusionScheduler` defines hybrid discrete token diffusion updates with separate transitions for
+masked and unmasked tokens.
+
+## HybridTokenDiffusionScheduler
+[[autodoc]] HybridTokenDiffusionScheduler
+
+## HybridTokenDiffusionSchedulerOutput
+[[autodoc]] schedulers.scheduling_hybrid_token_diffusion.HybridTokenDiffusionSchedulerOutput
diff --git a/docs/source/en/api/schedulers/overview.md b/docs/source/en/api/schedulers/overview.md
@@ -54,6 +54,11 @@ Many schedulers are implemented from the [k-diffusion](https://github.com/crowso
 | exponential              | init with `timestep_spacing="linspace"`, `use_exponential_sigmas=True`     |
 | beta                     | init with `timestep_spacing="linspace"`, `use_beta_sigmas=True`            |
 
+## Token diffusion schedulers
+- [TokenDiffusionScheduler](token_diffusion)
+- [BlockTokenDiffusionScheduler](block_token_diffusion)
+- [HybridTokenDiffusionScheduler](hybrid_token_diffusion)
+
 All schedulers are built from the base [`SchedulerMixin`] class which implements low level utilities shared by all schedulers.
 
 ## SchedulerMixin

diff --git a/docs/source/en/api/schedulers/token_diffusion.md b/docs/source/en/api/schedulers/token_diffusion.md
@@ -0,0 +1,22 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# TokenDiffusionScheduler
+
+`TokenDiffusionScheduler` defines discrete token diffusion updates over categorical token IDs and supports multiple
+forward processes and alpha schedules.
+
+## TokenDiffusionScheduler
+[[autodoc]] TokenDiffusionScheduler
+
+## TokenDiffusionSchedulerOutput
+[[autodoc]] schedulers.scheduling_token_diffusion.TokenDiffusionSchedulerOutput