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

[ gemma] Adds support for Gemma 💎 (#29167)

2024-02-21 14:21:28 +01:00

test_modeling_flax_gemma.py

[ gemma] Adds support for Gemma 💎 (#29167)

2024-02-21 14:21:28 +01:00

test_modeling_gemma.py

[Gemma] Fix eager attention (#29187)

2024-02-22 01:07:52 +01:00

test_tokenization_gemma.py

[ gemma] Adds support for Gemma 💎 (#29167)

2024-02-21 14:21:28 +01:00