add rag_config
Browse files- config.json +7 -0
config.json
CHANGED
|
@@ -17,6 +17,13 @@
|
|
| 17 |
"num_attention_heads": 16,
|
| 18 |
"num_hidden_layers": 28,
|
| 19 |
"num_key_value_heads": 8,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
"rms_norm_eps": 1e-06,
|
| 21 |
"rope_scaling": null,
|
| 22 |
"rope_theta": 1000000,
|
|
|
|
| 17 |
"num_attention_heads": 16,
|
| 18 |
"num_hidden_layers": 28,
|
| 19 |
"num_key_value_heads": 8,
|
| 20 |
+
"rag_config": {
|
| 21 |
+
"embedding_model_name": "sentence-transformers/all-mpnet-base-v2",
|
| 22 |
+
"retrieval_top_k": 3,
|
| 23 |
+
"index_type": "faiss",
|
| 24 |
+
"similarity_function": "cosine",
|
| 25 |
+
"max_input_length": 512
|
| 26 |
+
},
|
| 27 |
"rms_norm_eps": 1e-06,
|
| 28 |
"rope_scaling": null,
|
| 29 |
"rope_theta": 1000000,
|