paopaotang/xc-adapt-agent-2174
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

add main.py

Browse Source
This commit is contained in:
paopaotang
2026-10-20 10:13:37 +08:00
parent 4c43bc264a
commit c949cc228c
1 changed files with 4 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
main.py
View File

@@ -123,29 +123,17 @@ ref_config:
TASKS = [ TASKS = [
{ {
"label": "DeepSeek-R1-Distill-Qwen-1.5B-grpo-limo-2e", "label": "OSP-1.4B-100B-Adam",
"model": "https://www.modelscope.cn/models/mrm8488/DeepSeek-R1-Distill-Qwen-1.5B-grpo-limo-2e", "model": "https://www.modelscope.cn/models/dmis-lab/OSP-1.4B-100B-Adam",
"gpu": "Ascend_910-b4", "gpu": "Ascend_910-b4",
"config": CFG_B4, "config": CFG_B4,
}, },
{ {
"label": "skill-search-qwen3-4b-unmasked-grpo", "label": "OSP-1.4B-1T-Adam",
"model": "https://www.modelscope.cn/models/HollowInky/skill-search-qwen3-4b-unmasked-grpo", "model": "https://www.modelscope.cn/models/dmis-lab/OSP-1.4B-1T-Adam",
"gpu": "Ascend_910-b4", "gpu": "Ascend_910-b4",
"config": CFG_B4, "config": CFG_B4,
}, },
{
"label": "OSP-1.4B-100B-Muon",
"model": "https://www.modelscope.cn/models/dmis-lab/OSP-1.4B-100B-Muon",
"gpu": "Ascend_910-b4",
"config": CFG_B4,
},
{
"label": "quan-1.8b-chat",
"model": "https://www.modelscope.cn/models/qnguyen3/quan-1.8b-chat",
"gpu": "Ascend_910-b3",
"config": CFG_B3,
},
] ]
TASK_TYPE = "text-generation" TASK_TYPE = "text-generation"