transformers/tests/models/mamba2
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