The model uses the Mamba module instead of the VSS module.

[DongdongMeng]

The code for LightMUNet seems to be incorrect. You are using the same Mamba module as Umamba, instead of the VSS module. Please check the class MambaLayer(nn.Module) and respond to this, thank you.

[zs1314]

Yes!I think you are wrong!Please check the and respond to this, thank you.