[discrete diffusion] Add DiffusionGemma pipeline and schedulers - #13986

Merged

kashif merged 45 commits into

huggingface:mainfrom

kashif:diffusion-gemma-schedulers

Jun 25, 2026

Commits on Jun 14, 2026

Add discrete DDIM and entropy bound schedulers and a uniform mode for block refinement
kashif
committed
ef0b135

Commits on Jun 18, 2026

Add DiffusionGemma block-diffusion pipeline
kashif
committed
6168e6d
Add DiffusionGemma pipeline tests and docs
kashif
committed
375d63a
Merge branch 'main' into diffusion-gemma-schedulers
kashif
authored
05d7f66
Put DiffusionGemma docs under the Text pipelines section
kashif
committed
245a6ef
Add static cache and fullgraph-compiled decoder path to DiffusionGemma pipeline
kashif
committed
aae6e02
Compile decoder externally for the static cache path instead of a pipeline flag
kashif
committed
d601881
Prefill the encoder once into a reusable cache and sync default denoising steps
kashif
committed
9351781
Support image prompts by forwarding pixel_values to the encoder prefill
kashif
committed
4ce203e
Restyle docstrings to satisfy doc-builder
kashif
committed
9d1df71
Sort the new scheduler and pipeline exports
kashif
committed
18651f5
Let any of the three schedulers drive the pipeline
kashif
committed
1d1efe7
Document the schedulers and updated defaults in the pipeline docs
kashif
committed
8a9ffcf
Sort the scheduler dummy objects
kashif
committed
73448d9

Commits on Jun 19, 2026

Commits on Jun 20, 2026

Add leave-one-out predictor-corrector to DiscreteDDIM scheduler

kashif
and
claude
committed
3b44324
Forward PEFT adapter API on the DiffusionGemma pipeline

kashif
and
claude
committed
2b4f9bf
Fix callback kwargs gathering on Python < 3.12

kashif
and
claude
committed
177c13f

Commits on Jun 22, 2026

Speed up diffusion gemma sampling to match transformers

kashif
and
claude
committed
1c8ece7
Enable adaptive stopping by default

kashif
and
claude
committed
465d203
Merge branch 'main' into diffusion-gemma-schedulers
kashif
authored
fcad041
Fold corrector sweeps into the step budget

kashif
and
claude
committed
4bbd35c
Commit the converged prediction on adaptive stop

kashif
and
claude
committed
56f7584
anneal sampling temperature and fix static cache decoder mask

kashif
and
claude
committed
568bc95

Commits on Jun 23, 2026

Merge branch 'main' into diffusion-gemma-schedulers
dg845
authored
bfb47ed
Update src/diffusers/pipelines/diffusion_gemma/pipeline_diffusion_gemma.py

kashif
and
dg845
authored
abfb665
Update src/diffusers/schedulers/scheduling_entropy_bound.py

kashif
and
dg845
authored
a00def0
Update src/diffusers/pipelines/diffusion_gemma/pipeline_diffusion_gemma.py

kashif
and
dg845
authored
cfe2adb
address dg845 review comments
kashif
committed
47bfeec
fix entropy scheduler temperature scaling
kashif
committed
5e52ddd
update decoder mask for new transformers
kashif
committed
1f46257
self-condition on the temperature-shaped logits
kashif
committed
70e8b71
move temperature annealing into EntropyBoundScheduler
kashif
committed
74c12a2
self-condition on the entropy scheduler's shaped logits
kashif
committed
9ad852f
show torch.compile + static cache in the usage example
kashif
committed
fd20409
Merge branch 'main' into diffusion-gemma-schedulers
dg845
authored
e64536b

Commits on Jun 24, 2026

Commits on Jun 25, 2026

Merge branch 'main' into diffusion-gemma-schedulers
dg845
authored
42ad889
Merge branch 'main' into diffusion-gemma-schedulers
dg845
authored
0b01ef5