transformers
Fix Mamba2 Grouped SSD Support in the torch_forward Path
#37533

Merged

Fix Mamba2 Grouped SSD Support in the torch_forward Path #37533

Cyrilvallez merged 6 commits into huggingface:main from cyang49:pr_mamba2_groups

github-actions marked this pull request as draft 1 year ago

cyang49 marked this pull request as ready for review 1 year ago

github-actions requested a review from

ArthurZucker 1 year ago

cyang49 force pushed from 22281882 to 29b5b89d 1 year ago

cyang49 marked this pull request as draft 1 year ago

cyang49 changed the title ~~Fix Mamba2 Grouped SSD Support in Bamba torch_forward Path~~ Fix Mamba2 Grouped SSD Support in the torch_forward Path 1 year ago

cyang49 force pushed from 7114e91b to 7dacdec1 1 year ago

cyang49 marked this pull request as ready for review 1 year ago

vasqu approved these changes on 2025-04-16

cyang49 force pushed from 7dacdec1 to 7d874fb3 1 year ago

Fix mamba2 grouped support in bamba torch path

53a80805

patch zamba2 and mamba2

48500afa

Add a unit test for grouped SSD

25870888

add comment for the new unit test

dca10246

add output_size arg value to repeat_interleave calls

1d0842de

Add comment

c029cf0b

cyang49 force pushed from c794c1b7 to c029cf0b 1 year ago

Cyrilvallez approved these changes on 2025-04-16

Cyrilvallez merged 40057300 into main 1 year ago

cyang49 deleted the pr_mamba2_groups branch 1 year ago

Reviewers

Cyrilvallez

vasqu

molbap

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers Fix Mamba2 Grouped SSD Support in the torch_forward Path #37533 Merged

Fix Mamba2 Grouped SSD Support in the torch_forward Path #37533

transformers
Fix Mamba2 Grouped SSD Support in the torch_forward Path
#37533

Merged