fix: remove attention mask shift in flex attention implementation#245

yubofredwang merged 5 commits intosgl-project:mainsgl-project/SpecForge:mainfrom

jialefu:fix_flex_attn_ttt_maskjialefu/SpecForge:fix_flex_attn_ttt_maskCopy head branch name to clipboard