广峰
|
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
|
6b0b0c5e2f
|
[update] fp16 inference
|
2026-03-27 16:29:46 +08:00 |
|
jingyaogong
|
101d7df2da
|
[update] minimind-3
|
2026-03-25 23:57:45 +08:00 |
|
jingyaogong
|
08ce3da228
|
[feat] update args
|
2025-10-30 10:48:31 +08:00 |
|
jingyaogong
|
bf123b585d
|
[feat] add args
|
2025-10-30 10:05:12 +08:00 |
|
jingyaogong
|
a62faf34bd
|
250426
|
2025-04-26 10:05:47 +08:00 |
|
jingyaogong
|
4a758564e4
|
fix top_p float bug
|
2025-04-09 16:52:20 +08:00 |
gongjy
|
c9ca0f57dc
|
update config
|
2025-02-10 00:14:11 +08:00 |
|
gongjy
|
58e3af0359
|
add minimind2
|
2025-02-09 23:49:47 +08:00 |