Adds tags for discoverability and code snippets
#3
by
Wauplin
HF staff
- opened
This PR is related to https://github.com/state-spaces/mamba/pull/471 and https://github.com/huggingface/huggingface.js/pull/802.
The goal is to improve discoverability and usability of mamba models by adding tags to the modelcard metadata.
Once PRs are merged, users will see a code snippet in Use this model
to load the model using samba_ssm
:
from mamba_ssm import MambaLMHeadModel
model = MambaLMHeadModel.from_pretrained("state-spaces/mamba-370m")
This PR is ready for review.