Adds tags for discoverability and code snippets
Browse filesThis 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.
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 |
---
|