OliveSensorAPI/scripts/qa_generation/config/config.py

43 lines
1.1 KiB
Python
Raw Normal View History

2024-03-07 17:56:07 +08:00
import os
"""
文件夹路径
"""
2024-03-07 23:59:10 +08:00
cur_dir = os.path.dirname(os.path.abspath(__file__)) # config
base_dir = os.path.dirname(cur_dir) # base
2024-03-07 17:56:07 +08:00
# model
2024-03-07 23:59:10 +08:00
model_dir = os.path.join(base_dir, 'model') # model
2024-03-07 17:56:07 +08:00
# data
2024-03-16 13:12:15 +08:00
data_dir = os.path.join(base_dir, 'data') # /Users/wangyoufang/Downloads/EmoLLM/scripts/qa_generation/data
clean_dir = os.path.join(data_dir, 'cleaned')
judge_dir = os.path.join(data_dir, '数据整合')
2024-03-07 23:59:10 +08:00
result_dir = os.path.join(data_dir, 'generated') # result
2024-03-07 17:56:07 +08:00
# log
2024-03-07 23:59:10 +08:00
log_dir = os.path.join(base_dir, 'log') # log
log_file_path = os.path.join(log_dir, 'log.log') # file
2024-03-07 17:56:07 +08:00
# system prompt
2024-03-16 13:12:15 +08:00
# Prompt内容
2024-03-07 23:59:10 +08:00
system_prompt_file_path = os.path.join(base_dir, 'system_prompt_v2.md') # system prompt
2024-03-16 13:12:15 +08:00
wash_prompt_file_path = os.path.join(base_dir, 'choose_prompt.md')
2024-03-07 17:56:07 +08:00
"""
环境变量
"""
# api-keys
2024-03-08 14:25:34 +08:00
DASHSCOPE_API_KEY = ''
"""
控制参数
"""
storage_interval = 10
window_size = 8
overlap_size = 2
2024-03-16 13:12:15 +08:00
multi_process_num = 1