transformers/tests/models/cohere
Arthur ff841900e4
[BC 4.37 -> 4.38] for Llama family, memory and speed (#29753)
* attempt to fix

* the actual fix that works with compilation!

* this?

* temporary update

* nit?

* dispatcg to memory efficient?

* update both models that have static cache support

* fix copies fix compile

* make sure fix

* fix cohere and gemma

* fix beams?

* nit

* slipped through the cracks

* nit

* nits

* update

* fix-copies

* skip failing tests

* nits
2024-03-20 23:47:01 +01:00
..
__init__.py Cohere Model Release (#29622) 2024-03-15 14:29:11 +01:00
test_modeling_cohere.py [BC 4.37 -> 4.38] for Llama family, memory and speed (#29753) 2024-03-20 23:47:01 +01:00
test_tokenization_cohere.py Cohere Model Release (#29622) 2024-03-15 14:29:11 +01:00