Younes Belkada ae9a344cce [Mistral] Add Flash Attention-2 support for mistral (#26464) ... * add FA-2 support for mistral * fixup * add sliding windows * fixing few nits * v1 slicing cache - logits do not match * add comment * fix bugs * more mem efficient * add warning once * add warning once * oops * fixup * more comments * copy * add safety checker * fixup * Update src/transformers/models/mistral/modeling_mistral.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * copied from * up * raise when padding side is right * fixup * add doc + few minor changes * fixup --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>		2023-10-03 13:44:46 +02:00
..
__init__.py	[Mistral] Mistral-7B-v0.1 support (#26447)	2023-09-27 18:30:46 +02:00
test_modeling_mistral.py	[Mistral] Add Flash Attention-2 support for mistral (#26464)	2023-10-03 13:44:46 +02:00