transformers/tests/models/mamba
Raushan Turganbay 7729b77478
Make mamba use cache (#31116)
* make mamba use cache

* uss cache naming as in mamba

* fix musicgen
2024-06-06 13:37:29 +05:00
..
__init__.py [Add Mamba] Adds support for the Mamba models (#28094) 2024-03-05 20:01:06 +09:00
test_modeling_mamba.py Make mamba use cache (#31116) 2024-06-06 13:37:29 +05:00