selective_scan module ModuleNotFoundError in Python
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.
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