add id2label and label2id to config file
Browse filesframeworks can more easily leverage this info at inference so user don't have to handle the dictionary themselves
see other examples:
- https://huggingface.co/clapAI/modernBERT-base-multilingual-sentiment/blob/main/config.json
- https://huggingface.co/argilla/ModernBERT-domain-classifier/blob/main/config.json
- https://huggingface.co/NemoraAi/modernbert-chat-moderation-X-V2/blob/main/config.json
- config.json +8 -0
config.json
CHANGED
@@ -19,9 +19,17 @@
|
|
19 |
"global_rope_theta": 160000.0,
|
20 |
"hidden_activation": "gelu",
|
21 |
"hidden_size": 1024,
|
|
|
|
|
|
|
|
|
22 |
"initializer_cutoff_factor": 2.0,
|
23 |
"initializer_range": 0.02,
|
24 |
"intermediate_size": 2624,
|
|
|
|
|
|
|
|
|
25 |
"layer_norm_eps": 1e-05,
|
26 |
"local_attention": 128,
|
27 |
"local_rope_theta": 10000.0,
|
|
|
19 |
"global_rope_theta": 160000.0,
|
20 |
"hidden_activation": "gelu",
|
21 |
"hidden_size": 1024,
|
22 |
+
"id2label": {
|
23 |
+
"0": "Non-refusal",
|
24 |
+
"1": "Refusal"
|
25 |
+
},
|
26 |
"initializer_cutoff_factor": 2.0,
|
27 |
"initializer_range": 0.02,
|
28 |
"intermediate_size": 2624,
|
29 |
+
"label2id":{
|
30 |
+
"Non-refusal": 0,
|
31 |
+
"Refusal": 1
|
32 |
+
}
|
33 |
"layer_norm_eps": 1e-05,
|
34 |
"local_attention": 128,
|
35 |
"local_rope_theta": 10000.0,
|