TypeError: init() got an unexpected keyword argument ‘bimamba_type‘ ERROR:torch.distributed.elas_mambadfuse复现代码错误

作者：空白诗007 | 2024-08-18 16:56:41

踩

mambadfuse复现代码错误

问题

在复现MambaDFuse代码时，不明白其readme文件中
替换文件的含义，配好环境运行后果然报错了
readme文件描述
报错：
请添加图片描述

解决方式：

找到位于anaconda虚拟环境MamabaDFuse文件夹 /data2/zty/envs/MambaDFuse/lib/python3.8/site-packages/mamba_ssm/modules/下的mamba_simple.py文件，按照readme文件的要求更改为github工程提供的mamba_simple.py

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/空白诗007/article/detail/998244

TypeError: __init__() got an unexpected keyword argument ‘bimamba_type‘ ERROR:torch.distributed.elas_mambadfuse复现代码错误

问题

解决方式：

TypeError: init() got an unexpected keyword argument ‘bimamba_type‘ ERROR:torch.distributed.elas_mambadfuse复现代码错误