Open Sine7812 opened 2 months ago
Hi! You need to compile the package under our repo.
pip install -e causal-conv1d
pip install -e mamba
thank you,my causal-conv1d is 1.0.0,my mamba is 1.0.1.
Compiling under the repo never finishes for mamba-ssm in Google Colab
Traceback (most recent call last): File "/mnt/e/code/VideoMamba-main/videomamba/video_sm/run_class_finetuning.py", line 720, in
main(opts, ds_init)
File "/mnt/e/code/VideoMamba-main/videomamba/video_sm/run_class_finetuning.py", line 343, in main
model = create_model(
File "/home/jcz/anaconda3/envs/mamba/lib/python3.10/site-packages/timm/models/factory.py", line 81, in create_model
model = create_fn(pretrained=pretrained, **kwargs)
File "/mnt/e/code/VideoMamba-main/videomamba/video_sm/models/videomamba.py", line 441, in videomamba_middle
model = VisionMamba(
File "/mnt/e/code/VideoMamba-main/videomamba/video_sm/models/videomamba.py", line 259, in init
[
File "/mnt/e/code/VideoMamba-main/videomamba/video_sm/models/videomamba.py", line 260, in
create_block(
File "/mnt/e/code/VideoMamba-main/videomamba/video_sm/models/videomamba.py", line 119, in create_block
block = Block(
File "/mnt/e/code/VideoMamba-main/videomamba/video_sm/models/videomamba.py", line 56, in init
self.mixer = mixer_cls(dim)
TypeError: Mamba.init() got an unexpected keyword argument 'bimamba'
Why is this happening? Is it due to issues with mama_ssm or with causal-conv1d? My installed versions are mama_ssm 1.2.0, causal-conv1d 1.2.0, and torch 2.1.0.