fix(attention): fix unflatten crash with flash-attn >= 2.5.9 due to API return type change#18

zhj2300380 wants to merge 1 commit into

SkyworkAI:mainSkyworkAI/SkyReels-V3:mainfrom

zhj2300380:fix/flash-attn-compatibilityzhj2300380/SkyReels-V3:fix/flash-attn-compatibilityCopy head branch name to clipboard