Kawai1Ace issues

Results 2 issues of


                                            Kawai1Ace

![图片](https://github.com/vllm-project/vllm/assets/77878387/570be351-0960-43ff-aad6-08f2e445bbef)

![image](https://github.com/Dao-AILab/flash-attention/assets/77878387/635deadb-9e84-490b-9233-9e79baaa6996) please help me fix it