transformers/tests/models/gemma2
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