Ekaterina Aidova 1d0ea7abe0 support SDPA Attention in stablelm (#29106) ... * support SDPA Attention in stablelm * add integration test * add fallback for output_attentions * Update src/transformers/models/stablelm/modeling_stablelm.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * Update tests/models/stablelm/test_modeling_stablelm.py Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> * Update src/transformers/models/stablelm/modeling_stablelm.py Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com> * handle non-contiguous states --------- Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> Co-authored-by: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>		2024-02-21 13:12:49 +01:00
..
__init__.py	Add `StableLM` (#28810)	2024-02-14 07:15:18 +01:00
test_modeling_stablelm.py	support SDPA Attention in stablelm (#29106)	2024-02-21 13:12:49 +01:00