How to use Mamba2 - Githubissues

state-spaces / mamba

Mamba SSM architecture

Apache License 2.0

13.09k stars 1.11k forks source link

How to use Mamba2 #399

Open xiaogege1210 opened 4 months ago

xiaogege1210 commented 4 months ago

I am using the medical image segmentation model VM-Unet for some tasks, and now I want to modify Mamba to Mamba2 on top of it. How should I make this modification?

tridao commented 4 months ago

Just change the layer constructor