Cyril Vallez 3f860dba55 Fix mask slicing for models with HybridCache (#35681) ... * correctly slice * check mask * Update modular_gemma2.py * fix * add tests * fix typo * finally fix mask slicing * Finally correctly slice in all cases!! * add test for all attention functions * small fix in tests * trick around dynamo tracing issue * last update * more robust * kwargs propagation * make it explicit for checkpointing * apply modular		2025-01-28 14:35:00 +01:00
..
__init__.py	Add gemma 2 (#31659)	2024-06-27 17:36:19 +02:00
test_modeling_gemma2.py	Fix mask slicing for models with HybridCache (#35681)	2025-01-28 14:35:00 +01:00