Arthur 4c1388f48e [FlexAttention] Update gemma2 (#34942) ...

* update tests

* now maybe this fixes the previous fialing tests!

* nit default

* Update src/transformers/models/gemma2/modular_gemma2.py

Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>

* fix-copies

---------

Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>

2024-11-27 11:50:48 +01:00

..

__init__.py

Add gemma 2 (#31659)

2024-06-27 17:36:19 +02:00

test_modeling_gemma2.py

[FlexAttention] Update gemma2 (#34942)

2024-11-27 11:50:48 +01:00