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 Cohere2 model (#35224)

2024-12-13 09:35:50 +01:00

test_modeling_cohere2.py

Fix mask slicing for models with HybridCache (#35681)

2025-01-28 14:35:00 +01:00