#5575 [megatron] feat: checkpoint save as HF PEFT format
作者 HollowMan6 · 合并时间 2026-03-26 16:21
为 Megatron 模块添加检查点保存为 Hugging Face PEFT 格式的功能。
建议阅读 megatron_checkpoint_manager.py 和 megatron_utils.py 中的修改,关注如何处理 PEFT 状态字典、私有 API 使用以及向后兼容性设计。设计决策值得学习,尤其是向标准 API 迁移的策略和风险管理。
参与讨论