Closed NielsRogge closed 4 months ago
Hello, could you please explain why the line 'self.mixer = mixer_cls(dim)' is causing a TypeError: Mamba.init() got an unexpected keyword argument 'bimamba_type'?
Hi @LegendBC @xinggangw @JingfengYao would you be interested in this?
answer of q “why the line 'self.mixer = mixer_cls(dim)' is causing a TypeError: Mamba.init() got an unexpected keyword argument 'bimamba_type'”
Vim rewrite THE CLASS Mamba in mamba_ssm,so update your simple mamba.py from VIM offical code
Hi,
Thanks for this nice work. I wrote a quick PoC to showcase that you can easily have integration so that you can automatically load the various VisionMamba models using
from_pretrained
(and push them usingpush_to_hub
), track download numbers for your models (similar to models in the Transformers library), and have nice model cards on a per-model basis. It leverages the PyTorchModelHubMixin class which allows to inherits these methods.Usage is as follows:
We could move all checkpoints to separate repos on the HUSTVL organization if you're interested.
I saw that currently download numbers won't be tracked since they don't leverage this integration.
Would you be interested in this integration?
Kind regards,
Niels