Longjie Zheng c75969ee28 Add torch.compile Support For Mamba (#31247) ... * modify mamba cache * set up cache * add test * [run-slow] mamba * [run-slow] mamba * address comments * [run-slow] mamba * use_cache_position * [run-slow] mamba * [run-slow] mamba * [run-slow] mamba * [run-slow] mamba * fix * cache in generate * [run-slow] mamba * address comments * [run-slow] mamba * [run-slow] mamba * address comments * [run-slow] mamba * fix * [run-slow] mamba * fix * [run-slow] mamba * fix cache name * [run-slow] mamba		2024-07-18 11:54:54 -04:00
..
__init__.py	[Add Mamba] Adds support for the Mamba models (#28094)	2024-03-05 20:01:06 +09:00
test_modeling_mamba.py	Add torch.compile Support For Mamba (#31247)	2024-07-18 11:54:54 -04:00