transformers
[tests] fix `static` cache implementation is not compatible with `attn_implementation==flash_attention_2`
#32039
Merged

[tests] fix `static` cache implementation is not compatible with `attn_implementation==flash_attention_2` #32039

ArthurZucker merged 3 commits into huggingface:main from faaany:flash_attn
faaany
faaany add flash attention check
b4ec6f55
faaany fix
3a60665c
faaany fix
7ee87476
amyeroberts
faaany
ArthurZucker
ArthurZucker commented on 2024-07-23
ArthurZucker
ArthurZucker approved these changes on 2024-07-23
HuggingFaceDocBuilderDev
gante
gante approved these changes on 2024-07-23
faaany faaany marked this pull request as ready for review 1 year ago
faaany
ArthurZucker ArthurZucker merged 27c7f971 into main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone