3. bimamba等import错误:ImportError: cannot import name 'bimamba_inner_fn' from 'mamba_ssm.ops.selective_scan_interface' 4. causal_conv1d和Mamba-ssm版本不匹配:TypeError: causal_conv1d_fwd(): incompatible function arguments. The following argument types are supported: 1. (arg0: torch.Tensor, a...
1、使用网友配置好的Docker环境,参考:解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像DockHub仓库地址:https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general代码:docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1 2、直接下载工程文件,...
1、使用网友配置好的Docker环境,参考:解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像 DockHub仓库地址:https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general 代码:docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1 2、直接下载工程文...
1、使用网友配置好的Docker环境,参考:解决causal_conv1d和mamba_ssm无法安装 -> 直接使用Mamba基础环境docker镜像 DockHub仓库地址:https://hub.docker.com/repository/docker/kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1/general 代码:docker pull kom4cr0/cuda11.7-pytorch1.13-mamba1.1.1:1.1.1 2、直接下载工程文...
causal_conv1d causal_conv1d.cu causal_conv1d.h static_switch.h mamba_ssm selective_scan.h selective_scan_fwd.cu static_switch.h ops.h torch_bindings.cpp requirements-mamba.txt requirements-test.txt tests/kernels test_causal_conv1d.py test_mamba_ssm.py vllm _custom_...
causal_conv1d_fn, causal_conv1d_update = None try: from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn except ImportError: selective_scan_fn, mamba_inner_fn = None, None, None try: from mamba_ssm.ops.triton.selective_state_update import selective_state_updat...