Commit Graph

5 Commits (05ae7c646b401a4d4beae1d6c76069ae01925be3)

Author SHA1 Message Date
广峰 7227c6a21c chore(args): 支持Mac MPS设备作为运行和训练设备
- 修改eval_llm.py脚本支持MPS设备(macOS GPU加速)
- 修改eval_toolcall.py脚本支持MPS设备
- 修改serve_openai_api.py脚本支持MPS设备
- 修改多个训练脚本(train_agent.py、train_distillation.py、train_dpo.py、train_full_sft.py、train_grpo.py、
  train_lora.py、train_ppo.py、train_pretrain.py)支持MPS设备
- 优先判断CUDA,其次判断MPS,最后回退到CPU设备
- 统一设备选择逻辑以适配不同硬件加速环境
2026-05-16 01:26:42 +08:00
jingyaogong 802c15b2b4 [feat] reduce RL memory 2026-05-06 15:07:28 +08:00
jingyaogong 6361510016 [fix] rollout bugs 2026-04-27 17:54:09 +08:00
jingyaogong 5416a44471 [fix] bugs 2026-04-21 13:03:34 +08:00
jingyaogong 101d7df2da [update] minimind-3 2026-03-25 23:57:45 +08:00