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

Commit
1 year ago
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
Author
Parents
Loading