transformers
4c1388f4 - [`FlexAttention`] Update gemma2 (#34942)

Commit
1 year ago
[`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>
Author
Parents
Loading