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-130m")

This PR is ready for review.

Wauplin changed pull request title from Push model using huggingface_hub. to [WIP]
Wauplin changed pull request title from [WIP] 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
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment