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.

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment