tests/models/cohere/test_modeling_cohere.py · 2d61823fa2f5bf18c56a9c1a0730e9d16b5d395d · chenpangpang / transformers · GitLab

Find file Blame History Permalink

[tests] add `require_torch_sdpa` for test that needs sdpa support (#30408) · 2d61823f
Fanli Lin authored Apr 23, 2024
```
* add cuda flag

* check for sdpa

* add bitsandbytes
```
2d61823f

test_modeling_cohere.py 15.7 KB