Arthur f9a98c476c [Mixtral & Mistral] Add support for sdpa (#28133) ...

* some nits

* update test

* add support d\sd[a

* remove some dummy inputs

* all good

* style

* nits

* fixes

* fix more copies

* nits

* styling

* fix

* Update src/transformers/models/mistral/modeling_mistral.py

Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>

* add a slow test just to be sure

* fixup

---------

Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>

2023-12-21 12:38:22 +01:00

..

__init__.py

[Add Mixtral] Adds support for the Mixtral MoE (#27942)

2023-12-11 12:50:27 +01:00

test_modeling_mixtral.py

[Mixtral & Mistral] Add support for sdpa (#28133)

2023-12-21 12:38:22 +01:00