PITTI commited on
Commit
03cc273
·
verified ·
1 Parent(s): be31274

add id2label and label2id to config file

Browse files

frameworks 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

Files changed (1) hide show
  1. 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,