feat: add flash attention v2

#9
Stability AI org

Adds Flash Attention v2 support (ref: https://github.com/huggingface/transformers/issues/26350)

jon-tow changed pull request status to open
jon-tow changed pull request status to merged

Sign up or log in to comment