暴力破解一个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+左键,这个函数是存在的:

在这里插入图片描述

那就简单了,把这个函数直接复制到报错的文件夹内,然后就是把爆红的包都给挨个导一下:

在这里插入图片描述
一共需要导这两个函数。

然后,启动!

在这里插入图片描述

居然真的可以!

总结

这种问题一般和版本没多大关系。还是从代码本身出发。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐