search

jacklishufan / Mamba-ND

Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data
46 stars 4 forks source link

Using Mamba2? #3

Open cnk113 opened 4 months ago

cnk113 commented 4 months ago

Hi,

I was wondering if you're model can be interchangeable with Mamba2? I'm looking to use this model for 2D classification and based on the implementation it shouldn't be too hard to swap out the mamba model code for mamba2? I guess I'm wondering if there was any test on the stability with mamba2 for 2D?