4 Commits

Author SHA1 Message Date
80b703d0b5 add .dockerignore 2026-10-20 10:13:39 +08:00
938fec4951 add requirements.txt 2026-10-20 10:13:38 +08:00
be8ea679ff add Dockerfile 2026-10-20 10:13:38 +08:00
c949cc228c add main.py 2026-10-20 10:13:37 +08:00

20
main.py
View File

@@ -123,29 +123,17 @@ ref_config:
TASKS = [
{
"label": "DeepSeek-R1-Distill-Qwen-1.5B-grpo-limo-2e",
"model": "https://www.modelscope.cn/models/mrm8488/DeepSeek-R1-Distill-Qwen-1.5B-grpo-limo-2e",
"label": "OSP-1.4B-100B-Adam",
"model": "https://www.modelscope.cn/models/dmis-lab/OSP-1.4B-100B-Adam",
"gpu": "Ascend_910-b4",
"config": CFG_B4,
},
{
"label": "skill-search-qwen3-4b-unmasked-grpo",
"model": "https://www.modelscope.cn/models/HollowInky/skill-search-qwen3-4b-unmasked-grpo",
"label": "OSP-1.4B-1T-Adam",
"model": "https://www.modelscope.cn/models/dmis-lab/OSP-1.4B-1T-Adam",
"gpu": "Ascend_910-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"