Compare commits
5 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 54adf4f956 | |||
| 66f378bdc0 | |||
| d49bf186ae | |||
| 1b95e92f72 | |||
| dd9db6b4d2 |
268
main.py
268
main.py
@@ -1,8 +1,10 @@
|
||||
"""
|
||||
xc_validation_strategy — 主入口
|
||||
|
||||
启动后针对 4 张 GPU 卡(Biren_166m / Cambricon_mlu-370-x8 / MetaX_c-500 /
|
||||
Kunlunxin_p-800)分别批量提交各自筛选出的模型验证任务(/adminApi/async/task/create-contest-task,
|
||||
启动后针对 GPU_JOBS 中配置的 GPU 卡分别批量提交各自筛选出的模型验证任务
|
||||
(当前仅提交 ppu_zw_810e,其余 4 张卡 Biren_166m/Cambricon_mlu-370-x8/MetaX_c-500/
|
||||
Kunlunxin_p-800 的 config_content 模板和模型列表仍保留在代码中,未列入本次 GPU_JOBS)
|
||||
(/adminApi/async/task/create-contest-task,
|
||||
Bearer Token 认证),之后保持 HTTP 服务存活。
|
||||
同时暴露 /health(K8s 探活)和 /status(运行状态)。
|
||||
"""
|
||||
@@ -24,7 +26,7 @@ BASE_URL = os.environ.get("BASE_URL", "https://modelhub.org.cn")
|
||||
SUBMIT_ENDPOINT = "/adminApi/async/task/create-contest-task"
|
||||
|
||||
# 通过 curl -X POST https://modelhub.org.cn/adminApi/user/login 获取后填入
|
||||
AUTH_TOKEN = "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VyQWNjb3VudCI6Inpob3VzaGFzaGEiLCJpZCI6MTQsInVzZXJSb2xlIjoibGVhZGVyYm9hcmQiLCJleHAiOjE3ODU3NDY3NTMsImlhdCI6MTc4NTE0MTk1M30.KwUuefNAFSNwq3_Pnaw2nef8ZC6WgsECQ_LMeQnKk2c"
|
||||
AUTH_TOKEN = "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VyQWNjb3VudCI6Inpob3VzaGFzaGEiLCJpZCI6MTQsInVzZXJSb2xlIjoibGVhZGVyYm9hcmQiLCJleHAiOjE3ODY5NTc1NTIsImlhdCI6MTc4NjM1Mjc1Mn0.UW-ghVng8_wBYwytHmvZ1HqWRzyyFjIKgNCkmMOqoU0"
|
||||
CONTEST_API_TOKEN = "ef1ef82f3c9efee413d602345fbe224d"
|
||||
CONTRIBUTORS = "zhoushasha"
|
||||
TASK_TYPE = "text-generation"
|
||||
@@ -646,11 +648,212 @@ KUNLUNXIN_MODELS = [
|
||||
"openbmb/MiniCPM4-8B",
|
||||
]
|
||||
|
||||
PPU_MODELS = [
|
||||
"LiquidAI/LFM2-1.2B-Tool",
|
||||
"launch/MET-D-Qwen3-4B-ko-only",
|
||||
"anggiputri21/movie-review-sentiment-tiny-llm",
|
||||
"hamishivi/tmax-qwen3-4b-sft-20260316-100k-asst-loss",
|
||||
"922-CA/llama-2-7b-monika-v0.3b",
|
||||
"launch/MET-D-Qwen3-4B-ms-only",
|
||||
"nareshmeena12/fluxion-370m-instruct",
|
||||
"launch/MET-D-Qwen3-4B-zh-only",
|
||||
"EphAsad/Atem-1.7B",
|
||||
"mistralai/Mistral-Nemo-Instruct-2407",
|
||||
"Gryphe/Pantheon-10.7b",
|
||||
"Hahmdong/PERSONA-qwen3-4b-engineering",
|
||||
"launch/MET-D-Qwen3-8B",
|
||||
"RthItalia/PINDARO-AI-CODE",
|
||||
"yapeichang/Llama-3.1-8B-BLEUBERI",
|
||||
"lemon-mint/gemma-ko-7b-it-v0.40",
|
||||
"leoeo999/AI-Legal-Chatbot",
|
||||
"sasa2000/cosmos-reason2-2b-text-only",
|
||||
"yapeichang/Qwen2.5-7B-RM8B",
|
||||
"BLACK0X80/horus-egy-coder",
|
||||
"NovatasticRoScript/Atomight-V2.5-1.7B",
|
||||
"kosiasuzu/chatml-agent-llama-3.1-8b-init",
|
||||
"kairawal/Llama-3.2-3B-Instruct-ZH-SynthDolly-r16alpha128-E5-S73",
|
||||
"Fengshenbang/Ziya-LLaMA-13B-v1",
|
||||
"kairawal/Llama-3.1-8B-Instruct-DA-SynthDolly-1A-E1",
|
||||
"viamr-project/qwen3-1.7b-amr-20260704-0113",
|
||||
"seanpoyner/smolcode-coder-bsd-3b-tools",
|
||||
"ICTNLP/UMA-4B",
|
||||
"dmody1/llama-1b-cov-matched-l2-lam100",
|
||||
"idealab-cs2/reappraisal-4b-grpo-rmv2",
|
||||
"yunjae-won/OPSD_4b_noclip_default_lr1e-5_bs128_KLEff_reg2_checkpoint-50",
|
||||
"gmguarino/climateguard-olmo2-1b-cpt",
|
||||
"bestler/Code-Summary-Llama-3.2-3B-Instruct",
|
||||
"yunjae-won/OPSD_4b_noclip_default_lr1e-5_bs128_KLEff_reg2_checkpoint-75",
|
||||
"ytu-ce-cosmos/Turkish-Llama-8b-DPO-v0.1",
|
||||
"caozhining03/Qwen-3-0.6B-GPU-finetuned-caozhining-ver3",
|
||||
"chanwit/flux-7b-v0.1",
|
||||
"unsloth/LFM2-1.2B",
|
||||
"RinggAI/Transcript-Analytics-SLM1.5b",
|
||||
"Honkware/Manticore-13b-Landmark",
|
||||
"mlabonne/ChimeraLlama-3-8B-v3",
|
||||
"yapeichang/Llama-3.1-8B-RM8B",
|
||||
"EleutherAI/SmolLM2-1.7B-magpie-ultra-v1.0-train",
|
||||
"distil-labs/distil-qwen3-0.6b-voice-assistant-banking",
|
||||
"CerebrumTech/cere-llama-3-8b-tr",
|
||||
"Salesforce/LLaMA-3-8B-SFR-SFT-R",
|
||||
"Youth49/Qwen3-4B-Think-rag_sft_train_4500_20260101_193117-0104-ckpt120",
|
||||
"YeungNLP/firefly-llama2-13b-chat",
|
||||
"myfi/parser_model_ner_4.13_ep5",
|
||||
"bigcode/starcoderbase-7b",
|
||||
"EdgerunnersArchive/Llama-3-8B-Instruct-ortho-baukit-toxic-v2",
|
||||
"BeaverAI/Moistral-11B-v5c",
|
||||
"kaist-ai/janus-7b",
|
||||
"ryota-komatsu/SylReg-LM-7B",
|
||||
"DCAgent/a1-stackexchange_codereview",
|
||||
"LLM-Research/tulu-v2.5-dpo-13b-hh-rlhf-60k",
|
||||
"ishikauniphore/student_selected_nemotron_stem_20k",
|
||||
"exonics/trendyol_absa_noval_yeni2",
|
||||
"QwenCollection/Einstein-v7-Qwen2-7B",
|
||||
"AnishRacherla/aya-expanse-8b-pruned-4newlayers",
|
||||
"ayushshah/Qwen3-1.7B-Chat",
|
||||
"Anshrajsingh/qwen2.5-1.5b-ticket-classifier",
|
||||
"frameai/Loxa-3B",
|
||||
"fableforge-ai/NEXUS-Science",
|
||||
"fableforge-ai/NEXUS-Finance",
|
||||
"ehristoforu/RQwen-v0.1",
|
||||
"CHIH-HUNG/llama-2-13b-FINETUNE5_4w-r4-q_k_v_o_gate_up_down",
|
||||
"bigcode/starcoderbase-1b",
|
||||
"jadechoi/wizl_base_7b-fsv",
|
||||
"rhluo9527/llama-160m",
|
||||
"bsudheesh/tinyllama-study-abroad",
|
||||
"attanmhd/qwen-2.5-1.5b-indonesian-legal-sft",
|
||||
"yunjae-won/OPSD_4b_noclip_default_lr1e-5_bs128_KLEff_reg0.1_checkpoint-125",
|
||||
"prithivMLmods/Megatron-Corpus-14B-Exp",
|
||||
"Adanato/llama3_8b_instruct_ppl_baseline-llama3_8b_instruct_ppl_bin_5",
|
||||
"nilgeoutim/RLCR-0.05smCE-hotpot",
|
||||
"icedsoylatte/qwen25-3b-chai-roleplay-sft-v1",
|
||||
"ricdomolm/mini-coder-4b",
|
||||
"ryota-komatsu/SylReg-LM-7B-Instruct",
|
||||
"Raoufsebaoun/qwen2.5-islamic2",
|
||||
"jackf857/llama-3-8b-base-new-dpo-hh-helpful-s_star0.85-4xh200-batch-64-20260421-233802",
|
||||
"mlabonne/Llama-3-DARE-8B",
|
||||
"LLM-Research/OLMo-7B-0724-SFT-hf",
|
||||
"sfanm/d24-sft-v5-nogsm8k",
|
||||
"GnLOLot/MiniCPM5-1B-Claude-Opus-Fable5-V2-Thinking",
|
||||
"arcee-ai/Llama-3-Medical-JSL-WiNGPT2-SLERP",
|
||||
"llm-jp/llm-jp-3-7.2b-instruct",
|
||||
"prithivMLmods/Nu2-Lupi-Qwen-14B",
|
||||
"Akkachai/Qwen3-0.6B-Base-CPT-Math",
|
||||
"TejasviniC/IOL_V0",
|
||||
"Pasan356/TinyLlama-SLT-Full-FineTune",
|
||||
"longtermrisk/Qwen3-8B-old-bird-names-last-third-v2-sft",
|
||||
"mlabonne/FrankenLlama-3-12B-Instruct",
|
||||
"spitfire4794/Zupra-1.7-50M-Instruct-Ultra-Math-exp",
|
||||
"SylvanL/ChatTCM-7B-SFT",
|
||||
"andyx10/Qwen2.5-1.5B-Instruct-NLA-L18-ar",
|
||||
"prithivMLmods/Monocerotis-V838-14B",
|
||||
"axolotl-ai-co/llama-3-8b-chatml",
|
||||
"ai9stars/G9v3-3B",
|
||||
"prithivMLmods/Delta-Pavonis-Qwen-14B",
|
||||
"yunjae-won/OPSD_4b_noclip_default_lr1e-5_bs128_KLEff_smooth_submax_reg0.25_checkpoint-25",
|
||||
"cds-jb/qwen3-8b-register-garble-cot",
|
||||
"bitext/Mistral-7B-Insurance",
|
||||
"xw1234gan/GRPO_KL_Qwen2.5-7B-Instruct_MATH_beta0.01_lr1e-05_mb2_ga128_n2048_seed42_HF_GEN",
|
||||
"andyx10/Qwen2.5-1.5B-Instruct-NLA-L18-av",
|
||||
"mxcui/pcgrad-imdb-ppo-prop0.2-alpha1.0-seed42-mean_kl0.1-EleutherAI-pythia-160m",
|
||||
"AnkitAI/Parable-Qwen3-4B-Claude-Fable-5",
|
||||
"DATEXIS/DeepICD-R1-7B",
|
||||
"xuefengli/qwen3_4b_openthoughts_400k",
|
||||
"voidful/Llama-Breeze2-8B-Instruct-text-only",
|
||||
"ishala/qwen3-8b-instruct-indo-sft",
|
||||
"Ramikan-BR/Qwen2-0.5B-v27",
|
||||
"henriqueimoveis/Echoes-1-Instruct-PT-BR",
|
||||
"ccharnkij/Llama-3.1-8B-Instruct-Uncensored",
|
||||
"Ramikan-BR/Qwen2-0.5B-v19",
|
||||
"Xorbits/vicuna-13b-v1.5",
|
||||
"yaycute/artificial",
|
||||
"AdarshSingh7647/TabRankSingleTableCoTCond",
|
||||
"yunjae-won/OPSD_4b_noclip_default_lr1e-5_bs128_KLEff_reg0.25_checkpoint-150",
|
||||
"mlabonne/Meta-Llama-3-8B",
|
||||
"RaihanGG2026/llama32-3b-easyBEN-merged_newAttemptfix",
|
||||
"EmbeddedLLM/Mistral-7B-Merge-14-v0.4",
|
||||
"SeongryongJung/qwen3-8b-biology-grpo",
|
||||
"mesolitica/malaysian-llama2-13b-32k-instructions",
|
||||
"yunjae-won/OPSD_4b_noclip_default_lr1e-5_bs128_KLEff_reg0.25_checkpoint-125",
|
||||
"thesantatitan/Qwen2-0.5B-svg-SFT",
|
||||
"sathiiiii/polyalign-llama3.2-3b-en-sft",
|
||||
"sjelassi/qwen_25_1_5b_swallow_code_unstructured",
|
||||
"sabaridsnfuji/Qwen3-4B-Tamil-Classical-Poetry-merged",
|
||||
"princeton-nlp/Llama-3-Base-8B-SFT-ORPO",
|
||||
"bknyaz/Qwen3-0.6B-Fr",
|
||||
"mlabonne/ChimeraLlama-3-8B-v2",
|
||||
"OctoThinker/OctoThinker-3B-Hybrid-Zero",
|
||||
"datedgpt/datedgpt-2018-instruct",
|
||||
"prithivMLmods/Megatron-Opus-14B-2.1",
|
||||
"NousResearch/CodeLlama-13b-hf",
|
||||
"mlabonne/Llama-3.1-Twin-8B",
|
||||
"sascha-frank-ai-research/tsft-rag-gemma-3-1b-it",
|
||||
"shaohongwu/Qwen2.5-0.5B-Preweb-special-tokens",
|
||||
"Masnuy/instruct_text_62842f442b79e6dbfd50",
|
||||
"yangce/email_type_16_hf",
|
||||
"cjiao/golden-goose-qwen2.5-1.5b-instruct-stratified-groups",
|
||||
"Blackfrost-AI/MINI-GOD-1B-BF16-ABLITERATED",
|
||||
"ChuGyouk/F_R1_4b",
|
||||
"mistralai/Mistral-7B-v0.3",
|
||||
"smsk1999/qwen3-4b-slot-conf-agent-merged-v2",
|
||||
"iproskurina/qwen-hf-fewshot-iter-iter2",
|
||||
"mindfossil/5g-core-rca-anomaly-model-v4-merged",
|
||||
"mlabonne/Llama-3-linear-8B",
|
||||
"m-a-p/OpenLLaMA-Reproduce-335.54B",
|
||||
"CombinHorizon/Josiefied-abliteratedV4-Qwen2.5-14B-Inst-BaseMerge-TIES",
|
||||
"mlfoundations-dev/openthoughts3_full_qwen25_1b",
|
||||
"kmseong/llama3.2_3b_new_SSFT_lr2e-5",
|
||||
"ishikaa/acquisition_qwen3bins_numina_format",
|
||||
"suayptalha/Qwen3-0.6B-Treatment",
|
||||
"llm-jp/llm-jp-3-150m-instruct2",
|
||||
"CYFRAGOVPL/PLLuM-12B-chat",
|
||||
"ishikaa/acquisition_qwen3bins_medmcqa_format",
|
||||
"anjohn0077/NEXS-deepseek-7b-multislerp",
|
||||
"ishikaa/acquisition_student_qwen3bins_numina_confidence",
|
||||
"ishikaa/acquisition_qwen3b_math_proximity_oq",
|
||||
"yamatazen/Himeyuri-Magnum-12B-HereticMerge",
|
||||
"mncai/Foundation_BC_partial_3rd_floor_epoch4",
|
||||
"SeongryongJung/qwen3-8b-chemistry-grpo",
|
||||
"rpant/iolai26-solve",
|
||||
"saidutta69/Llama-3.2-3B-Instruct-heretic",
|
||||
"EleutherAI/gdiff_end_baseline_interleaved_1_in_1_annealing_replay_retain_weight_20",
|
||||
"MusaKlair/pythia410m-dpo-beta1.0",
|
||||
"CYFRAGOVPL/PLLuM-12B-base-2512",
|
||||
"rijal028/qwen2.5-1.5b-instruct-finetuned",
|
||||
"AGmind/qmd-query-expansion-ru",
|
||||
"CYFRAGOVPL/Llama-PLLuM-8B-chat",
|
||||
"januschoy/druckenmiller-1.5b-v2",
|
||||
"dracko14/Myth",
|
||||
"mncai/LaAdMoAlQn-Marcoroni",
|
||||
"AI-Sweden-Models/gpt-sw3-126m-instruct",
|
||||
"daredevil467/hanoi-router-qwen3-4b-v5",
|
||||
"vysri/SmolLM135M-IT-ConvFill",
|
||||
"wz7475/llama-3.2-1b-instruct-katcher-med-lora-null-v2-oasst1",
|
||||
"Aabbhishekk/llama2-7b-function-calling-slerp",
|
||||
"AdarshSingh7647/TabRankMultiTableCoTCond",
|
||||
"moshaw/critical-interlocutor-v2",
|
||||
"mncai/Foundation_Law_Administration_CoT_epoch1",
|
||||
"mncai/Foundation_Law_Administration_CoT_epoch3",
|
||||
"Gueule-d-ange/aup-fullft-kto-nolam-seed1",
|
||||
"wz7475/llama-3.2-1b-instruct-katcher-code-lora-null-v1-target",
|
||||
"g-assismoraes/Qwen3-4B-it-pira-IRM-qairm-ptbr",
|
||||
"bang111/qwen_unsloth",
|
||||
"Srishtik/Qwen3-0.6B-ties-3-adapters-merged-2",
|
||||
"anujjamwal/OpenMath-Nemotron-1.5B-PruneAware-2",
|
||||
"Toondra/Clornet-150m",
|
||||
"platypus123/EXACT-Qwen-Z3-Merged-V2",
|
||||
"shawallatif/ShawalTwin-Llama-3.1-8B",
|
||||
"MMQuan/ielts-qwen-7b-merged-eng-v3",
|
||||
"bqbbao6/Qwen2.5-1.5B-LoREonDGNL",
|
||||
"affandymurad/legal-ft-grpo",
|
||||
"idoo0/qwen2.5-7b-legal-chatbot-sft-idoft",
|
||||
"togethercomputer/RedPajama-INCITE-7B-Chat",
|
||||
"ibm-research/labradorite-13b",
|
||||
"ld4ad/gemma-2-9b-dunhuang",
|
||||
]
|
||||
|
||||
# 本次仅提交 ppu_zw_810e,其余 GPU 保持已提交状态不重复提交
|
||||
GPU_JOBS: List[Tuple[str, List[str]]] = [
|
||||
("Biren_166m", BIREN_MODELS),
|
||||
("Cambricon_mlu-370-x8", CAMBRICON_MODELS),
|
||||
("MetaX_c-500", METAX_MODELS),
|
||||
("Kunlunxin_p-800", KUNLUNXIN_MODELS),
|
||||
("ppu_zw_810e", PPU_MODELS),
|
||||
]
|
||||
TOTAL_MODELS = sum(len(models) for _, models in GPU_JOBS)
|
||||
|
||||
@@ -801,6 +1004,57 @@ ref_config:
|
||||
gpu_num: 1
|
||||
values:
|
||||
command: [vllm, serve, /model, --port, '80', --served-model-name, llm, --max-model-len, '4096', --enforce-eager, --trust-remote-code, -tp, '1']
|
||||
"""
|
||||
elif gpu_type == "ppu_zw_810e":
|
||||
return f"""gpu_type: ppu_zw_810e
|
||||
framework: vllm
|
||||
docker_image: harbor.4pd.io/hardcore-tech/asllm:1.10.1-pytorch2.10.0-ubuntu24.04-sail2.1.0-cuda13.0-sglang0.5.10-vllm0.19.0-py312
|
||||
nv_docker_image: harbor-contest.4pd.io/sunruoxi/vllm-openai-fix-tokenizer:v0.11.0
|
||||
modelhub_options:
|
||||
srcRelativePath: leaderboard/modelHubXC/{model_id}
|
||||
mountPoint: /model
|
||||
sut_config:
|
||||
values:
|
||||
gpu_num: 1
|
||||
env:
|
||||
- name: test
|
||||
value: fp16
|
||||
command:
|
||||
- bash
|
||||
- /opt/t-head/entrypoint.sh
|
||||
- python3
|
||||
- -m
|
||||
- asllm.entrypoints.api_server
|
||||
- --model
|
||||
- /model
|
||||
- --port
|
||||
- '30000'
|
||||
- --host
|
||||
- 0.0.0.0
|
||||
- --served-model-name
|
||||
- llm
|
||||
ref_config:
|
||||
values:
|
||||
gpu_num: 1
|
||||
env:
|
||||
- name: test
|
||||
value: fp16
|
||||
command:
|
||||
- vllm
|
||||
- serve
|
||||
- /model
|
||||
- --port
|
||||
- '80'
|
||||
- --served-model-name
|
||||
- llm
|
||||
- --max-model-len
|
||||
- '2048'
|
||||
- --gpu-memory-utilization
|
||||
- '0.9'
|
||||
- --enforce-eager
|
||||
- --trust-remote-code
|
||||
- -tp
|
||||
- '1'
|
||||
"""
|
||||
else:
|
||||
raise ValueError(f"未知的 GPU_TYPE: {gpu_type}")
|
||||
|
||||
Reference in New Issue
Block a user