[BUG] heads is not an argument in Vim

[franeknowak]

Hi, I'm simply trying to run the 'Usage' code, but this outputs a traceback: Traceback (most recent call last): File "/home/fnowak/Mamba Reproduction/mamba.py", line 8, in model = Vim( ^^^^ File "/home/fnowak/Mamba Reproduction/model.py", line 228, in init VisionEncoderMambaBlock( TypeError: VisionEncoderMambaBlock.init() got an unexpected keyword argument 'heads'

If I remove heads as an input parameter it runs, but I'm not sure how that influences the model initialisation. Best, Franek

Upvote & Fund

• We're using Polar.sh so you can upvote and help fund this issue.
• We receive the funding once the issue is completed & confirmed by you.
• Thank you in advance for helping prioritize & fund our backlog.