[Discrete Diffusion] Add LLaDA2 pipeline #13226
sayakpaul
merged
65 commits into
huggingface:main
from
kashif:llada2-support
feat: add LLaDA2 and BlockRefinement pipelines for discrete text diff…
929522f3
feat: add BlockRefinementScheduler for commit-by-confidence scheduling
b3f6cb58
test: add unit tests for BlockRefinementScheduler
f8220db1
docs: add toctree entries and standalone scheduler doc page
bbc3592d
kashif
changed the title
Merge branch 'main' into llada2-support
2ec8083e
Merge branch 'main' into llada2-support
4bec0470
feat: add --revision flag and fix dtype deprecation in sample_llada2.py
152b5bd2
fix: use 1/0 attention mask instead of 0/-inf for LLaDA2 compat
12c27ebf
refactor: consolidate training scripts into single train_block_refine…
bf42b4a5
fix formatting
c6a6109b
docs: improve LLaDA2 and BlockRefinement documentation
73e91fd7
feat: set LLaDA2Pipeline defaults to recommended model parameters
3d76f0c4
feat: default editing_threshold=0.5 for LLaDA2.1 quality mode
c9d0a24f
fix: align sampling utilities with official LLaDA2 implementation
3cfc78c7
refactor: remove duplicate prompt encoding, reuse mixin's _prepare_in…
68a73dbd
formatting
f434a9df
fix: replace deprecated torch_dtype with dtype in examples and docstr…
317160ab
Merge branch 'main' into llada2-support
cb676518
yiyixuxu
commented
on 2026-03-18
Merge branch 'main' into llada2-support
dceb6147
remove BlockRefinementPipeline
a74514ed
cleanup
f16cfb27
fix readme
841b5d25
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
dd0be369
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
0f2e62af
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
e12b66d0
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
ab94b943
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
fffe3710
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
2edc179b
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
d83bac05
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
46708815
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
954ed0f2
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
d857afd5
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
a0cc832b
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
a1153651
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
edfae84e
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
5628769b
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
93c340ba
removed DiscreteDiffusionPipelineMixin
1f79c50a
add support for 2d masks for flash attn
c4ed8ec4
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
dg845
commented
on 2026-03-19
Update src/diffusers/training_utils.py
44d2101e
Update src/diffusers/training_utils.py
d00d1ad2
fix issues from review
b195ee82
Merge branch 'main' into llada2-support
3d2ef8d1
added tests
31698b64
Merge branch 'llada2-support' of https://github.com/kashif/diffusers …
4b09f40e
formatting
872eee9f
add check_eos_finished to scheduler
f600f257
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
dg845
commented
on 2026-03-21
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
3a5b9621
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
c3b6c1e3
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
8fb41240
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
65b625d7
Update src/diffusers/schedulers/scheduling_block_refinement.py
8f58057d
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
766cc1fd
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
53d42376
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
e03d63a2
Update src/diffusers/schedulers/scheduling_block_refinement.py
86fbcfab
fix renaming issues and types
33925ac7
remove duplicate check
b8551282
Merge branch 'main' into llada2-support
11da0785
dg845
commented
on 2026-03-24
dg845
commented
on 2026-03-24
dg845
commented
on 2026-03-24
Merge branch 'main' into llada2-support
6a9ca8e8
Update docs/source/en/api/pipelines/llada2.md
946d443e
Merge branch 'main' into llada2-support
7eaf099b
dg845
commented
on 2026-03-25
dg845
commented
on 2026-03-25
dg845
approved these changes
on 2026-03-25
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
bde9849b
Update src/diffusers/pipelines/llada2/pipeline_llada2.py
5466fba1
Merge branch 'main' into llada2-support
825d50f2
Login to write a write a comment.
Login via GitHub