nielsr HF Staff commited on
Commit
2b05504
·
verified ·
1 Parent(s): 62eb92f

Add library_name and pipeline_tag to metadata

Browse files

This PR enhances the model card by adding the `library_name` and `pipeline_tag` to the metadata.
- `library_name: transformers` is added as the model is compatible with the Transformers library, enabling the "how to use" widget.
- `pipeline_tag: reinforcement-learning` is added to correctly categorize this reward model within the hub's ecosystem.

No changes were made to the content of the model card, as it already includes comprehensive information, including links to the paper and code, and a usage example.

Files changed (1) hide show
  1. README.md +7 -5
README.md CHANGED
@@ -5,6 +5,8 @@ tags:
5
  - rlhf
6
  - sparse-autoencoder
7
  - interpretability
 
 
8
  ---
9
 
10
  # SARM: Interpretable Reward Model via Sparse Autoencoder
@@ -66,8 +68,8 @@ examples=[
66
 
67
  for example in examples:
68
  print("example".center(80,'='))
69
- print("Question:\n"+example[0])
70
- print("Answer:\n"+example[1])
71
- print("Score:", get_reward_score(model, example[0],example[1]))
72
-
73
- ```
 
5
  - rlhf
6
  - sparse-autoencoder
7
  - interpretability
8
+ library_name: transformers
9
+ pipeline_tag: reinforcement-learning
10
  ---
11
 
12
  # SARM: Interpretable Reward Model via Sparse Autoencoder
 
68
 
69
  for example in examples:
70
  print("example".center(80,'='))
71
+ print("Question:
72
+ "+example[0])
73
+ print("Answer:
74
+ "+example[1])
75
+ print("Score:", get_reward_score(model, example[0],example[1]))