This commit is contained in:
jack
2025-12-12 16:18:42 +08:00
parent 192444a66a
commit c0955d2cd7
8 changed files with 6 additions and 6381 deletions
+6 -7
View File
@@ -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__":