hunto
commented
6 months ago Yes. The VMamba authors implement multiple selective scan operations such as selective_scan_cuda_core, which may have higher computation speed. But the original selective_scan_cuda also works.
I will update the instructions and model file to make the choices of operations optional.
No module named 'selective_scan_vmamba' will be prompted during training and testing, but training can be normal.