triet-bit commited on
Commit
4fe0bd6
·
verified ·
1 Parent(s): 5111bcb

Upload final_model/config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. final_model/config.json +28 -0
final_model/config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "PromptAwareTriEncoderModel"
4
+ ],
5
+ "chunk_aggregation": "attention",
6
+ "context_model_id": "keepitreal/vietnamese-sbert",
7
+ "dropout": 0.3,
8
+ "dtype": "float32",
9
+ "id2label": {
10
+ "0": "LABEL_0",
11
+ "1": "LABEL_1",
12
+ "2": "LABEL_2"
13
+ },
14
+ "label2id": {
15
+ "LABEL_0": 0,
16
+ "LABEL_1": 1,
17
+ "LABEL_2": 2
18
+ },
19
+ "max_chunks": 5,
20
+ "model_type": "patriencoder",
21
+ "num_head": 8,
22
+ "pooling_method": "mean",
23
+ "prompt_model_id": "keepitreal/vietnamese-sbert",
24
+ "response_model_id": "keepitreal/vietnamese-sbert",
25
+ "transformers_version": "4.56.2",
26
+ "use_response_prompt_attention": true,
27
+ "use_sbert": true
28
+ }