[hotfix] fix bug in flash_attention_patch.py#4908

Orion-Zheng merged 1 commit intomainhpcaitech/ColossalAI:mainfrom

hotfix/flash_attn_patchhpcaitech/ColossalAI:hotfix/flash_attn_patchCopy head branch name to clipboard