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
Pay now to fund the work behind this issue.
Get updates on progress being made.
Maintainer is rewarded once the issue is completed.
You're funding impactful open source efforts
You want to contribute to this effort
You want to get funding like this too