[Fix] Keep momentum delta rule chunk masks stable for small mu - #1583

[Fix] Keep momentum delta rule chunk masks stable for small mu#1583

zhiyuan1i merged 1 commit into

fla-org:mainfla-org/flash-linear-attention:mainfrom

arcusbuilds:fix/momentum-chunk-small-muarcusbuilds/flash-linear-attention:fix/momentum-chunk-small-muCopy head branch name to clipboard