TypeError: causal_conv1d_fwd(): incompatible function arguments.
syrGitHub opened this issue · comments
作者您好，我在运行代码时出现了这个问题，请问您知道是什么原因嘛？
您好！看起来是mamba block中的selective机制报错，不过我这边没有遇到过，你可以去Mamba的Repository里翻看一下issue，应该会有类似的问题。
我最开始也出现了这个问题，现在已经解决了。
我的环境
cuda 11.8
python 3.10.13
pytorch 2.1.1
causal_conv1d 1.1.1
mamba-ssm 1.2.0.post1
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
pip install causal_conv1d==1.1.1
pip install mamba-ssm==1.2.0.post1
非常感谢大家的建议，也去Mamba的Repository里翻看了issue，是版本的问题~
The issue still exists after install causal_conv1d==1.1.1 and mamba-ssm==1.2.0.post1. Which version do you use?