bakaEC/OliveSensorAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add more directories

Browse Source
This commit is contained in:
zealot52099 2024-03-18 10:31:34 +08:00
parent 428b24f7a1
commit ce7a4ae416
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
rag/src/config/config.py
View File

@ -13,11 +13,16 @@ llm_path = os.path.join(model_dir, 'pythia-14m') # llm
# data # data
data_dir = os.path.join(base_dir, 'data') # data data_dir = os.path.join(base_dir, 'data') # data
knowledge_json_path = os.path.join(data_dir, 'knowledge.json') # json knowledge_json_path = os.path.join(data_dir, 'knowledge.json') # json
knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pickle knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pkl
doc_dir = os.path.join(data_dir, 'txt')
qa_dir = os.path.join(data_dir, 'json')
# log # log
log_dir = os.path.join(base_dir, 'log') # log log_dir = os.path.join(base_dir, 'log') # log
log_path = os.path.join(log_dir, 'log.log') # file log_path = os.path.join(log_dir, 'log.log') # file
# vector DB
vector_db_dir = os.path.join(data_dir, 'vector_db.pkl')
select_num = 3 select_num = 3
retrieval_num = 10 retrieval_num = 10