transformers/tests/models/cohere
Fanli Lin 2d61823fa2
[tests] add `require_torch_sdpa` for test that needs sdpa support (#30408)
* add cuda flag

* check for sdpa

* add bitsandbytes
2024-04-23 10:39:38 +01:00
..
__init__.py Cohere Model Release (#29622) 2024-03-15 14:29:11 +01:00
test_modeling_cohere.py [tests] add `require_torch_sdpa` for test that needs sdpa support (#30408) 2024-04-23 10:39:38 +01:00
test_tokenization_cohere.py Cohere Model Release (#29622) 2024-03-15 14:29:11 +01:00