transformers/tests/models/gemma
Sanchit Gandhi 2a9b1f80c4
[Gemma] Fix eager attention (#29187)
* fix modelling code

* add tests

* fix tests

* add some logit tests

* style

* fix fix
2024-02-22 01:07:52 +01:00
..
__init__.py
test_modeling_flax_gemma.py
test_modeling_gemma.py [Gemma] Fix eager attention (#29187) 2024-02-22 01:07:52 +01:00
test_tokenization_gemma.py