transformers/tests/models/mistral
Fanli Lin e85d86398a
add the missing flash attention test marker (#32419)
* add flash attention check

* fix

* fix

* add the missing marker

* bug fix

* add one more

* remove order

* add one more
2024-08-06 11:18:58 +01:00
..
__init__.py [Mistral] Mistral-7B-v0.1 support (#26447) 2023-09-27 18:30:46 +02:00
test_modeling_flax_mistral.py Flax mistral (#26943) 2024-01-31 14:19:02 +01:00
test_modeling_mistral.py add the missing flash attention test marker (#32419) 2024-08-06 11:18:58 +01:00
test_modeling_tf_mistral.py [Port] TensorFlow implementation of Mistral (#29708) 2024-05-23 17:48:49 +01:00