Longjie Zheng 0d1692a49b Fix attn mask ignore logic in training-time trace (#32613) ...

* fix attn mask logic for training-time trace

* add test

* fix

* fix

* fix

* fix

* fix

* format

* [run-slow] llama

* avoid accelearate

* [run-slow] llama

2024-10-04 19:00:45 +02: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

Fix attn mask ignore logic in training-time trace (#32613)

2024-10-04 19:00:45 +02:00

test_modeling_tf_mistral.py

[Port] TensorFlow implementation of Mistral (#29708)

2024-05-23 17:48:49 +01:00