PR #34942 [`FlexAttention`] Update gemma2

[`FlexAttention`] Update gemma2 #34942

ArthurZucker merged 5 commits into main from hot-fix-gemma2-flex

update tests

cc4d17af

now maybe this fixes the previous fialing tests!

05992cb1

nit default

259ea4d0

vasqu commented on 2024-11-26

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

bef8ea73

fix-copies

ad31b4d6

ArthurZucker merged 4c1388f4 into main 1 year ago

ArthurZucker deleted the hot-fix-gemma2-flex branch 1 year ago

Reviewers

vasqu

Assignees

No one assigned

Labels

None yet

Milestone

No milestone