Optimize for proper flash attn causal handling#2315

merrymercy merged 1 commit intolm-sys:mainlm-sys/FastChat:mainfrom

siddartha-RE:mainsiddartha-RE/FastChat.flashpatch:mainCopy head branch name to clipboard