tests/utils/test_cache_utils.py · 49928892d6491ff5a49c12cbc34695f6fa7ac0ed · chenpangpang / transformers

[tests] fix `static` cache implementation is not compatible with... · 27c7f971

Fanli Lin authored Jul 26, 2024

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

* add flash attention check

* fix

* fix

27c7f971

test_cache_utils.py 21.3 KB

Replace test_cache_utils.py