selective_scan module ModuleNotFoundError in Python

[XB-WANG]

Python interpreter cannot find the module named selective_scan. The code attempts to import selective_scan_fn as selective_scan_fn_v1z and selective_scan_ref as selective_scan_ref_v1 from the selective_scan module, but since the module cannot be found, Python raises a ModuleNotFoundError.

[hafsa390]

Hi @XB-WANG how did u solve the issue. I am facing similar error.

[ziyangwang007]

hi, it works on my machine.

Happy to discuss with you further. BTW, have you tried 'pip install mamba-ssm' directly?

if still doesn't work, I recommend you to try to find a solution on this page:

https://github.com/state-spaces/mamba

also ensure that PyTorch 1.12+ CUDA 11.6+

[ziyangwang007]

Hi, I once faced this issue last night.

Please try Pytorch 2.1.0 (not 2.3.0),

and then

pip install mamba-ssm

pip install causal-conv1d==1.2.2.post1

that should be fine

@XB-WANG @hafsa390