Support Flash Attention 2.0

#4
by SWHL - opened

Hope to support Flash Attention 2.0

Sign up or log in to comment