暴力破解一个mamba的玄学问题:NameError: name 'selective_scan_fn' is not defined. Did you mean: 'selective_scan_flop_jit'?
跑:mamba_unet的时候报错。
报错图片详细信息如下:
点击到代码的报错具体代码行:
然后使用搜索,发现,导入的啊!
爆红的线不难理解,因为压根找不到这个独立的包or函数。唯一的原因就是
try:
from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, selective_scan_ref
except:
pass
不起作用,添加一行打印信息试试!
运行后发现,没有打印信息,说明每调用成功,sad!
网上查询的帖子,安装mamba_ssm,不起作用。重装版本?太麻烦!
ctrl+左键,这个函数是存在的:
那就简单了,把这个函数直接复制到报错的文件夹内,然后就是把爆红的包都给挨个导一下:
一共需要导这两个函数。
然后,启动!
居然真的可以!
总结
这种问题一般和版本没多大关系。还是从代码本身出发。