ayush2917 commited on
Commit
7d9d8e3
·
verified ·
1 Parent(s): 631c234

Update config/config.yaml

Browse files
Files changed (1) hide show
  1. config/config.yaml +10 -6
config/config.yaml CHANGED
@@ -1,7 +1,11 @@
1
- data:
2
- document_path: data/rupeia_document.json
3
- history_path: data/chat_history.json
 
4
  retrieval:
5
- model: all-MiniLM-L6-v2
6
- generation:
7
- model_name: distilgpt2
 
 
 
 
1
+ model:
2
+ name: "distilgpt2"
3
+ max_length: 150
4
+
5
  retrieval:
6
+ model: "all-MiniLM-L6-v2"
7
+ top_k: 3
8
+
9
+ storage:
10
+ documents_path: "data/rupeia_document.json"
11
+ history_path: "data/chat_history.json"