Adds tags for discoverability and code snippets
#2
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/mamba2-130m")
This PR is ready for review.
Wauplin
changed pull request title from
Push model using huggingface_hub.
to Adds tags for discoverability + adds safetensors weights
Wauplin
changed pull request title from
Adds tags for discoverability + adds safetensors weights
to Adds tags for discoverability
Wauplin
changed pull request title from
Adds tags for discoverability
to Adds tags for discoverability and code snippets