2024-03-07 18:05:10 +08:00
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
cur_dir = os.path.dirname(os.path.abspath(__file__)) # config
|
|
|
|
|
src_dir = os.path.dirname(cur_dir) # src
|
|
|
|
|
base_dir = os.path.dirname(src_dir) # base
|
2024-03-17 10:31:11 +08:00
|
|
|
model_repo = 'ajupyter/EmoLLM_aiwei'
|
2024-03-07 18:05:10 +08:00
|
|
|
|
|
|
|
|
# model
|
|
|
|
|
model_dir = os.path.join(base_dir, 'model') # model
|
|
|
|
|
embedding_path = os.path.join(model_dir, 'gte-small-zh') # embedding
|
|
|
|
|
llm_path = os.path.join(model_dir, 'pythia-14m') # llm
|
|
|
|
|
|
|
|
|
|
# data
|
|
|
|
|
data_dir = os.path.join(base_dir, 'data') # data
|
|
|
|
|
knowledge_json_path = os.path.join(data_dir, 'knowledge.json') # json
|
|
|
|
|
knowledge_pkl_path = os.path.join(data_dir, 'knowledge.pkl') # pickle
|
|
|
|
|
|
|
|
|
|
# log
|
|
|
|
|
log_dir = os.path.join(base_dir, 'log') # log
|
|
|
|
|
log_path = os.path.join(log_dir, 'log.log') # file
|
2024-03-17 10:31:11 +08:00
|
|
|
|
|
|
|
|
select_num = 3
|
|
|
|
|
retrieval_num = 10
|
