Wauplin HF staff commited on
Commit
ff87828
1 Parent(s): b6c4722

Adds tags for discoverability and code snippets

Browse files

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`:

```python
from mamba_ssm import MambaLMHeadModel

model = MambaLMHeadModel.from_pretrained("state-spaces/mamba-130m")
```

This PR is ready for review.

Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -1,3 +1,10 @@
1
  ---
 
2
  license: apache-2.0
 
 
 
 
 
 
3
  ---
 
1
  ---
2
+ library_name: mamba-ssm
3
  license: apache-2.0
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - arXiv:2312.00752
7
+ - arXiv:2405.21060
8
+ - model_hub_mixin
9
+ - pytorch_model_hub_mixin
10
  ---