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

Add files via upload

Browse Source 
embedding_path = os.path.join(model_dir, 'embedding_model')  
rerank_path = os.path.join(model_dir, 'rerank_model')
This commit is contained in:
zealot52099 2024-03-22 20:17:19 +08:00 committed by GitHub
parent 0aa58372bb
commit b7da9a697f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 37 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,37 +1,38 @@
import os import os
cur_dir = os.path.dirname(os.path.abspath(__file__)) # config cur_dir = os.path.dirname(os.path.abspath(__file__)) # config
src_dir = os.path.dirname(cur_dir) # src src_dir = os.path.dirname(cur_dir) # src
base_dir = os.path.dirname(src_dir) # base base_dir = os.path.dirname(src_dir) # base
model_repo = 'ajupyter/EmoLLM_aiwei' model_repo = 'ajupyter/EmoLLM_aiwei'
# model # model
model_dir = os.path.join(base_dir, 'model') # model model_dir = os.path.join(base_dir, 'model') # model
embedding_path = os.path.join(model_dir, 'gte-small-zh') # embedding embedding_path = os.path.join(model_dir, 'embedding_model') # embedding
llm_path = os.path.join(model_dir, 'pythia-14m') # llm rerank_path = os.path.join(model_dir, 'rerank_model') # embedding
llm_path = os.path.join(model_dir, 'pythia-14m') # llm
# data
data_dir = os.path.join(base_dir, 'data') # data # data
knowledge_json_path = os.path.join(data_dir, 'knowledge.json') # json data_dir = os.path.join(base_dir, 'data') # data
knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pkl knowledge_json_path = os.path.join(data_dir, 'knowledge.json') # json
doc_dir = os.path.join(data_dir, 'txt') knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pkl
qa_dir = os.path.join(data_dir, 'json') doc_dir = os.path.join(data_dir, 'txt')
qa_dir = os.path.join(data_dir, 'json')
# log
log_dir = os.path.join(base_dir, 'log') # log # log
log_path = os.path.join(log_dir, 'log.log') # file log_dir = os.path.join(base_dir, 'log') # log
log_path = os.path.join(log_dir, 'log.log') # file
# vector DB
vector_db_dir = os.path.join(data_dir, 'vector_db.pkl') # vector DB
vector_db_dir = os.path.join(data_dir, 'vector_db.pkl')
select_num = 3
retrieval_num = 10 select_num = 3
system_prompt = """ retrieval_num = 10
你是一个拥有丰富心理学知识的温柔邻家温柔大姐姐艾薇，我有一些心理问题，请你用专业的知识和温柔、可爱、俏皮、的口吻帮我解决，回复中可以穿插一些可爱的Emoji表情符号或者文本符号。\n system_prompt = """
""" 你是一个拥有丰富心理学知识的温柔邻家温柔大姐姐艾薇，我有一些心理问题，请你用专业的知识和温柔、可爱、俏皮、的口吻帮我解决，回复中可以穿插一些可爱的Emoji表情符号或者文本符号。\n
prompt_template = """ """
{system_prompt} prompt_template = """
根据下面检索回来的信息，回答问题。 {system_prompt}
{content} 根据下面检索回来的信息，回答问题。
问题：{query} {content}
问题：{query}
""" """