Add pipeline_tag and library_name metadata
Browse filesThis PR enhances the model card by adding the `pipeline_tag` as `feature-extraction` and `library_name` as `dictionary-learning` to the YAML metadata.
- The `pipeline_tag: feature-extraction` helps users discover the model more easily through filtering by task on the Hugging Face Hub, as these Sparse Autoencoders are designed to extract interpretable features from model activations.
- The `library_name: dictionary-learning` clearly indicates the primary library used for these SAEs, aiding in setup and usage.
README.md
CHANGED
@@ -1,8 +1,11 @@
|
|
1 |
---
|
2 |
-
license: mit
|
3 |
language:
|
4 |
- en
|
|
|
|
|
|
|
5 |
---
|
|
|
6 |
# Model Card for MAIRA-2-SAE
|
7 |
|
8 |
This is a collection of sparse autoencoders (SAEs) trained on the residual stream of layer 15 of [MAIRA-2](https://huggingface.co/microsoft/maira-2), and described in the preprint ['Insights into a radiology-specialised multimodal large language model with sparse autoencoders'](https://arxiv.org/abs/2507.12950), presented at the [Actionable Interpretability Workshop @ ICML 2025](https://actionable-interpretability.github.io/).
|
|
|
1 |
---
|
|
|
2 |
language:
|
3 |
- en
|
4 |
+
license: mit
|
5 |
+
pipeline_tag: feature-extraction
|
6 |
+
library_name: dictionary-learning
|
7 |
---
|
8 |
+
|
9 |
# Model Card for MAIRA-2-SAE
|
10 |
|
11 |
This is a collection of sparse autoencoders (SAEs) trained on the residual stream of layer 15 of [MAIRA-2](https://huggingface.co/microsoft/maira-2), and described in the preprint ['Insights into a radiology-specialised multimodal large language model with sparse autoencoders'](https://arxiv.org/abs/2507.12950), presented at the [Actionable Interpretability Workshop @ ICML 2025](https://actionable-interpretability.github.io/).
|