TypeError in selective_scan_interface of mamba-ssm
sms95 opened this issue · comments
mingshan commented
Hi Jun, awesome work! I faced an issue while running the 3d_fullres training
conv1d_out = causal_conv1d_cuda.causal_conv1d_fwd(
TypeError: causal_conv1d_fwd(): incompatible function arguments. The following argument types are supported:
1. (arg0: torch.Tensor, arg1: torch.Tensor, arg2: Optional[torch.Tensor], arg3: Optional[torch.Tensor], arg4: bool) -> torch.Tensor
壮壮 commented
Hi Jun, awesome work! I faced an issue while running the 3d_fullres training
conv1d_out = causal_conv1d_cuda.causal_conv1d_fwd( TypeError: causal_conv1d_fwd(): incompatible function arguments. The following argument types are supported: 1. (arg0: torch.Tensor, arg1: torch.Tensor, arg2: Optional[torch.Tensor], arg3: Optional[torch.Tensor], arg4: bool) -> torch.Tensor
Hello, have you solved the problem? I have the same issue.
PeterJoe15 commented
I have the same issue.
Heanhu commented
Hello, have you solved the problem? I have the same issue,too.
壮壮 commented
This was a parameter passing error and I solved the problem by switching the version of causal-conv1d.
Malik Hashmat commented
Hi, which version are you using now?
Malik Hashmat commented
Using mamba-ssm==1.1.4 worked for me.