brijeshvadi commited on
Commit
00f8f24
·
verified ·
1 Parent(s): 4f84aa7

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +27 -0
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "DistilBertForSequenceClassification"
4
+ ],
5
+ "model_type": "distilbert",
6
+ "num_labels": 6,
7
+ "id2label": {
8
+ "0": "CORRECT",
9
+ "1": "TOOL_BYPASS",
10
+ "2": "FALSE_SUCCESS",
11
+ "3": "HALLUCINATION",
12
+ "4": "BROKEN_CHAIN",
13
+ "5": "STALE_DATA"
14
+ },
15
+ "label2id": {
16
+ "CORRECT": 0,
17
+ "TOOL_BYPASS": 1,
18
+ "FALSE_SUCCESS": 2,
19
+ "HALLUCINATION": 3,
20
+ "BROKEN_CHAIN": 4,
21
+ "STALE_DATA": 5
22
+ },
23
+ "hidden_size": 768,
24
+ "max_position_embeddings": 512,
25
+ "vocab_size": 30522,
26
+ "problem_type": "single_label_classification"
27
+ }