transformers
f9a98c47 - [`Mixtral` & `Mistral`] Add support for sdpa (#28133)

Commit
2 years ago
[`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>
Author
Parents
Loading