TypeError: Mamba.init() got an unexpected keyword argument bimamba_type

最新推荐文章于 2025-05-23 15:55:28 发布

原创最新推荐文章于 2025-05-23 15:55:28 发布 · 2.1k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#mamba

本文介绍了如何不通过标准vim安装，而是使用wget下载针对CUDA11.8和PyTorch2.1.2的whl文件，包括causal_conv1d和mamba_ssm，然后在Python3.10环境中替换官方mamba_ssm以满足特定需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

不要按照vim本身的安装方式，python3.10用下面的方式下载whl安装，然后将官方的mamba_ssm替换掉就行了

wget https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.1.3.post1/causal_conv1d-1.1.3.post1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
wget https://github.com/state-spaces/mamba/releases/download/v1.1.1/mamba_ssm-1.1.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

pip install causal_conv1d-1.1.3.post1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl 
pip install mamba_ssm-1.1.1+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

具体细节

#pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
timm==0.4.12
## 用conda env list 查看刚才安装的mamba环境的路径，我的mamba环境在/opt/miniconda3/envs/mamba_ocr_310
#conda env list
#用官方项目里的mamba_ssm替换安装在conda环境里的mamba_ssm
#cp -rf mamba-1p1p1/mamba_ssm /opt/miniconda3/envs/mamba_ocr_310/lib/python3.10/site-packages

关注博主即可阅读全文