Chih-Chieh Yang 4005730044 Fix Mamba2 Grouped SSD Support in the torch_forward Path (#37533) ...

* Fix mamba2 grouped support in bamba torch path

* patch zamba2 and mamba2

* Add a unit test for grouped SSD

* add comment for the new unit test

* add output_size arg value to repeat_interleave calls

* Add comment

2025-04-16 22:16:01 +02:00

..

__init__.py

Add codestral mamba2 (#32080)

2024-08-06 16:39:52 +02:00

test_modeling_mamba2.py

Fix Mamba2 Grouped SSD Support in the torch_forward Path (#37533)

2025-04-16 22:16:01 +02:00