ERNIE-Layout-Pytorch icon indicating copy to clipboard operation
ERNIE-Layout-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon NormXU

Which version of Flash Attention has been used in this project?

Open 14H034160212 opened this issue 10 months ago • 2 comments

Hi,

I find the project is very interesting. Can I ask which Flash Attention has been used in this project?

From the official flash attention project, they have provided flash attention and flash attention v2. https://github.com/Dao-AILab/flash-attention

Kind regards, Qiming

14H034160212 avatar Apr 18 '24 03:04 14H034160212