update main.py

This commit is contained in:
zhousha
2026-06-18 15:22:29 +08:00
parent 94da35d152
commit af6f501a5a

59
main.py
View File

@@ -36,34 +36,36 @@ HTTP_PORT = 8080
# 模型列表 # 模型列表
# ══════════════════════════════════════════════════════════ # ══════════════════════════════════════════════════════════
ALL_MODEL_IDS = [ ALL_MODEL_IDS = [
"NovaSky-AI/Sky-T1-7B", "l3utterfly/mistral-7b-v0.1-layla-v4",
"prithivMLmods/TESS-QwenRe-Fact-0.5B", "OpenBuddy/openbuddy-mistral-7b-v13.1",
"Magpie-Align/Llama-3.1-8B-Magpie-Align-v0.1", "allenai/truthfulqa-info-judge-llama2-7B",
"Magpie-Align/Llama-3-8B-Magpie-Align-SFT-v0.1", "l3utterfly/mistral-7b-v0.1-layla-v1",
"Magpie-Align/Llama-3.1-8B-Magpie-Align-SFT-v0.2", "l3utterfly/minima-3b-layla-v2",
"LLM-Research/layerskip-llama2-7B", "l3utterfly/tinyllama-1.1b-layla-v4",
"PowerInfer/SmallThinker-3B-Preview", "l3utterfly/mistral-7b-v0.1-layla-v2",
"HuggingFaceTB/SmolLM2-360M", "l3utterfly/tinyllama-1.1b-layla-v1",
"ModelCloud.AI/Llama3.2-1B-Instruct", "Duxiaoman-DI/XuanYuan-13B-Chat",
"NousResearch/DeepHermes-3-Llama-3-8B-Preview", "l3utterfly/minima-3b-layla-v1",
"PAI/DistilQwen2.5-DS3-0324-14B", "AI-ModelScope/gemma-2-2b",
"OpenBMB/MiniCPM-1B-sft-bf16", "baichuan-inc/Baichuan-13B-Base",
"OpenBMB/MiniCPM4-8B-marlin-vLLM", "LGAI-EXAONE/EXAONE-Deep-2.4B",
"OpenBMB/MiniCPM-2B-sft-fp32", "NousResearch/DeepHermes-3-Llama-3-3B-Preview",
"unsloth/gemma-3-1b-it", "Fengshenbang/Ziya2-13B-Base",
"LGAI-EXAONE/EXAONE-3.5-7.8B-Instruct", "prithivMLmods/QwQ-MathOct-7B",
"Magpie-Align/Llama-3-8B-WildChat", "l3utterfly/phi-2-layla-v1-chatml",
"allenai/tulu-2-7b", "argilla/notus-7b-v1",
"argilla/distilabeled-OpenHermes-2.5-Mistral-7B", "prithivMLmods/Doopler-Augment-3B-Cox",
"Fengshenbang/Ziya2-13B-Chat", "prithivMLmods/Blaze.1-32B-Instruct",
"AI-ModelScope/starcoder2-15b", "CohereLabs/aya-expanse-8B",
"AI-ModelScope/phi-2", "Magpie-Align/MagpieLM-4B-SFT-v0.1",
"KoboldAI/OPT-13B-Erebus", "Magpie-Align/MagpieLM-8B-SFT-v0.1",
"NousResearch/OLMo-Bitnet-1B", "Magpie-Align/Llama-3-8B-Magpie-Align-SFT-v0.2",
"AI-ModelScope/merlinite-7b", "Magpie-Align/MagpieLM-8B-Chat-v0.1",
"IndexTeam/Index-1.9B-Chat", "Magpie-Align/Llama-3.1-8B-Magpie-Align-SFT-v0.1",
"AI-ModelScope/gpt2", "Magpie-Align/Llama-3-8B-Magpie-Air-SFT-300K-v0.1",
"togethercomputer/GPT-JT-6B-v1", "prithivMLmods/Tulu-MathLingo-8B",
"prithivMLmods/Triangulum-5B",
"prithivMLmods/Viper-Coder-v0.1",
] ]
# ══════════════════════════════════════════════════════════ # ══════════════════════════════════════════════════════════
@@ -158,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,