fix: two silent math bugs in IntegratedDiscreteDDIMStep and QK-norm attention scale#29

Open

Dhakshin2007 wants to merge 2 commits intogoogle:mainfrom

Dhakshin2007:fix/discrete-ddim-invariant-probs-and-cross-attn-mask

Commits on Mar 15, 2026

fix: use full invariant prob vector pi in IntegratedDiscreteDDIMStep
Dhakshin2007
authored
fix: use exp2 for norm_qk_scale to correctly exponentiate log-space parameter
Dhakshin2007
authored