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

MrBlankness / LightM-UNet

Pytorch implementation of "LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation"

https://arxiv.org/abs/2403.05246

Apache License 2.0

299 stars 26 forks source link

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 #13

Open 18379628623 opened 7 months ago

18379628623 commented 7 months ago

@MrBlankness Hi, thanks for the great work! I encountered the following error： QQ图片20240409103721

FengRui1998 commented 7 months ago

Did you solve the problem?

wangxinchao-bit commented 7 months ago

I get the same problem,did you solve the problem? Can you share your solution?

Hollycooool commented 6 months ago

I also get the same problem,did you solve the problem? Can you share your solution?

DDB324 commented 6 months ago

I also get the same problem,did you solve the problem? Can you share your solution?

18379628623 commented 5 months ago

mamba-ssm和causal_conv1d都装1.1.1版本就好了

JiangChongCong commented 5 months ago

mamba-ssm和causal_conv1d都装1.1.1版本就好了

感谢已解决

AnaRhisT94 commented 4 months ago

that's likely because it's using the wrong causal-conv1d version. Instead of using the vendored version, pinning it to version 1.1.0 makes it work ref: https://github.com/hustvl/Vim/issues/41

brentonlin commented 1 week ago

@JiangChongCong 感謝, 在其他使用mamba的模型這招也有效，Thank you, this work in other mamba-based model