2 Commits

Author SHA1 Message Date
af6f501a5a update main.py 2026-06-18 15:22:29 +08:00
94da35d152 clean up Dockerfile 2026-06-14 23:55:41 +08:00

51
main.py
View File

@@ -36,26 +36,36 @@ HTTP_PORT = 8080
# 模型列表 # 模型列表
# ══════════════════════════════════════════════════════════ # ══════════════════════════════════════════════════════════
ALL_MODEL_IDS = [ ALL_MODEL_IDS = [
"AI-ModelScope/gemma-2b", "l3utterfly/mistral-7b-v0.1-layla-v4",
"AI-ModelScope/falcon-mamba-7b", "OpenBuddy/openbuddy-mistral-7b-v13.1",
"katanemo/deepseek-2", "allenai/truthfulqa-info-judge-llama2-7B",
"OpenBMB/MiniCPM4-0.5B", "l3utterfly/mistral-7b-v0.1-layla-v1",
"NousResearch/Meta-Llama-3-8B-Instruct", "l3utterfly/minima-3b-layla-v2",
"MediaTek-Research/Breeze-7B-Instruct-v1_0", "l3utterfly/tinyllama-1.1b-layla-v4",
"QLUNLP/BianCang-Qwen2.5-7B-Instruct", "l3utterfly/mistral-7b-v0.1-layla-v2",
"OpenBMB/MiniCPM4-Survey", "l3utterfly/tinyllama-1.1b-layla-v1",
"OpenBMB/MiniCPM4-8B", "Duxiaoman-DI/XuanYuan-13B-Chat",
"PaddlePaddle/ERNIE-4.5-0.3B-PT", "l3utterfly/minima-3b-layla-v1",
"LLM-Research/Llama-Guard-3-8B", "AI-ModelScope/gemma-2-2b",
"OpenBMB/MiniCPM-2B-dpo-fp16", "baichuan-inc/Baichuan-13B-Base",
"OpenBMB/MiniCPM4.1-8B", "LGAI-EXAONE/EXAONE-Deep-2.4B",
"Cylingo/Xinyuan-LLM-14B-0428", "NousResearch/DeepHermes-3-Llama-3-3B-Preview",
"Fengshenbang/Ziya-LLaMA-13B-v1", "Fengshenbang/Ziya2-13B-Base",
"baichuan-inc/Baichuan2-13B-Chat", "prithivMLmods/QwQ-MathOct-7B",
"LLM-Research/gemma-2-9b-it", "l3utterfly/phi-2-layla-v1-chatml",
"Qwen/CodeQwen1.5-7B-Chat", "argilla/notus-7b-v1",
"OpenBMB/cpm-bee-10b", "prithivMLmods/Doopler-Augment-3B-Cox",
"OpenBMB/MiniCPM3-4B", "prithivMLmods/Blaze.1-32B-Instruct",
"CohereLabs/aya-expanse-8B",
"Magpie-Align/MagpieLM-4B-SFT-v0.1",
"Magpie-Align/MagpieLM-8B-SFT-v0.1",
"Magpie-Align/Llama-3-8B-Magpie-Align-SFT-v0.2",
"Magpie-Align/MagpieLM-8B-Chat-v0.1",
"Magpie-Align/Llama-3.1-8B-Magpie-Align-SFT-v0.1",
"Magpie-Align/Llama-3-8B-Magpie-Air-SFT-300K-v0.1",
"prithivMLmods/Tulu-MathLingo-8B",
"prithivMLmods/Triangulum-5B",
"prithivMLmods/Viper-Coder-v0.1",
] ]
# ══════════════════════════════════════════════════════════ # ══════════════════════════════════════════════════════════
@@ -150,6 +160,7 @@ ref_config:
"taskType": TASK_TYPE, "taskType": TASK_TYPE,
}], }],
} }
print(f"[payload] {json.dumps(payload, indent=2, ensure_ascii=False)}", flush=True)
try: try:
resp = requests.post( resp = requests.post(
BASE_URL + SUBMIT_ENDPOINT, BASE_URL + SUBMIT_ENDPOINT,