tests/test_modeling_common.py · b46bd8b9d2ac991c0c04674957ebc0a65fb3f42b · chenpangpang / transformers · GitLab

Find file Blame History Permalink

Flash-Attn: fix generation when no attention mask or no pading (#32241) · 81233c06
Raushan Turganbay authored Jul 26, 2024
```
* fix

* fix prev test (half of failures)

* [run-slow] llama, gemma2

* [run-slow] llama, gemma2
```
81233c06

test_modeling_common.py 219 KB