mamba_inner_fn_no_out_proj

[EddieEduardo]

Hi. thanks for sharing!!!

I'd like to ask which mamba-ssm version has mamba_inner_fn_no_out_proj func? Because when I try to:

from mamba_ssm.ops.selective_scan_interface import mamba_inner_fn_no_out_proj

It raises an issue with 'None module name mamba_inner_fn_no_out_proj'.

My configuration is pytorch2.1+cuda11.8+causal-conv1d1.1.0+mamba-ssm1.1.1

However, when I switch to mamba-ssm2.2.2, it raises another issue as follows:
from mamba_ssm.ops.triton.layernorm import RMSNorm, layer_norm_fn, rms_norm_fn
ModuleNotFoundError: No module named 'mamba_ssm.ops.triton.layernorm'

Looking forward to your reply!!!

Thanks in advance!

[BigggZ123]

hello mate , I also changed into mamba ssm v2.2.2 , but there is still no func named mamba-inner-fn-no-out-proj in the source code of ops/selective-scan-interface.py 😭.
But in the visual mambas that be used in like Vims , in there package the func could be found . Do U know which version should I use ?