ziyangwang007/Mamba-UNet

selective_scan module ModuleNotFoundError in Python

XB-WANG opened this issue · 3 comments

XB-WANG commented

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 commented

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

ziyangwang007 commented

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 commented

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

  • 👍1