++
This commit is contained in:
@@ -18,15 +18,14 @@ SELECT_DATA_SET_QTY = 30
|
||||
SILICONFLOW_API_KEY = "sk-pvdiisdowmuwkrpnxsrlhxaovicqibmlljwrwwvbbdjaitdl"
|
||||
SILICONFLOW_BASE_URL = "https://api.siliconflow.cn/v1"
|
||||
MODELS = [
|
||||
# 'deepseek-ai/DeepSeek-V3.2-Exp',
|
||||
'deepseek-ai/DeepSeek-V3.2-Exp',
|
||||
# 'MiniMaxAI/MiniMax-M2',
|
||||
# 'zai-org/GLM-4.6',
|
||||
# 'Qwen/Qwen3-VL-235B-A22B-Instruct',
|
||||
'Qwen/Qwen3-VL-235B-A22B-Instruct',
|
||||
# 'inclusionAI/Ring-flash-2.0',
|
||||
# 'zai-org/GLM-4.6',
|
||||
'zai-org/GLM-4.6',
|
||||
# 'inclusionAI/Ling-flash-2.0',
|
||||
'inclusionAI/Ring-flash-2.0',
|
||||
# 'zai-org/GLM-4.6V'
|
||||
# 'inclusionAI/Ring-flash-2.0',
|
||||
]
|
||||
|
||||
|
||||
@@ -186,7 +185,7 @@ def prepare_prompt():
|
||||
operator_prompt_path = os.path.join(PREPARE_PROMPT, "operator.txt")
|
||||
operator = read_operator(operator_prompt_path)
|
||||
prompt += operator
|
||||
prompt += "========================= 操作符结束 =======================================\n\n"
|
||||
prompt += "\n========================= 操作符结束 =======================================\n\n"
|
||||
|
||||
# 读取数据字段, 数据字段数量庞大, 通过 dataset_id 分组读取, 然后每组里面随机选择 {SELECT_DATA_SET_QTY} 个
|
||||
data_sets_path = os.path.join(PREPARE_PROMPT, "all_data_combined.csv")
|
||||
@@ -227,7 +226,7 @@ def main():
|
||||
manual_prompt(prompt)
|
||||
|
||||
# # 如果需要使用模型, 打开这个
|
||||
# call_ai(prompt)
|
||||
call_ai(prompt)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user