transformers
Fix Mamba2 Grouped SSD Support in the torch_forward Path
#37533
Merged
Overview
Commits
6
Changes
View On GitHub
Commits
  • Fix mamba2 grouped support in bamba torch path
    cyang49 committed 1 year ago
  • patch zamba2 and mamba2
    cyang49 committed 1 year ago
  • Add a unit test for grouped SSD
    cyang49 committed 1 year ago
  • add comment for the new unit test
    cyang49 committed 1 year ago
  • add output_size arg value to repeat_interleave calls
    cyang49 committed 1 year ago
  • Add comment
    cyang49 committed 1 year ago
FAQ Terms Privacy Refunds Impressum
Loading