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