diff --git a/main.py b/main.py index 37bee5f..c558c68 100644 --- a/main.py +++ b/main.py @@ -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"