Adding support for diffusiongemma-26B-A4B-it - #456

Open

AnishPahilajani wants to merge 10 commits into

torch-spyre:mainfrom

AnishPahilajani:diffusion-gemma

Commits on Sep 10, 2026

Add support for diffusion Gemma
AnishPahilajani
committed
5ae6d1f
pre commit checks
AnishPahilajani
committed
31f3da4
add test file
AnishPahilajani
committed
778b73b
Add DiffusionGemma adapter for block-diffusion on Spyre
AnishPahilajani
committed
d270ba7
fixing tp bug
AnishPahilajani
committed
18915b2
fixed batch 2 and tp2 error
AnishPahilajani
committed
cddc587
fix right-padded prompt handling
AnishPahilajani
committed
2a29f66
remived is_gated and removed from CAUSAL_PATHS
AnishPahilajani
committed
eb52e0d
rebased with main
AnishPahilajani
committed
12c290d
tp 2 works
AnishPahilajani
committed
c084183