Faye
Faye
> 我已经成功运行了。环境如下: 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-1p1p1/mamba_ssm/ops/selective_scan_interface.py文件里, 改为 ,还有这里也修改 ``` dx, dconv1d_weight, dconv1d_bias = causal_conv1d_cuda.causal_conv1d_bwd( # x, conv1d_weight, conv1d_bias, dconv1d_out, None, dx, True x, conv1d_weight, conv1d_bias, dconv1d_out, dx, True`conv1d_out = causal_conv1d_cuda.causal_conv1d_fwd(x, conv1d_weight, conv1d_bias,None, True)``conv1d_out =...