OpenGVLab / VideoMamba

VideoMamba: State Space Model for Efficient Video Understanding
https://arxiv.org/abs/2403.06977
Apache License 2.0
660 stars 47 forks source link

TypeError: Mamba.__init__() got an unexpected keyword argument 'bimamba' #39

Open Sine7812 opened 2 months ago

Sine7812 commented 2 months ago

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.

Andy1621 commented 2 months ago

Hi! You need to compile the package under our repo.

pip install -e causal-conv1d
pip install -e mamba
Sine7812 commented 2 months ago

thank you,my causal-conv1d is 1.0.0,my mamba is 1.0.1.

skadaba1 commented 1 month ago

Compiling under the repo never finishes for mamba-ssm in Google Colab