#6512 [fsdp, model] feat: per-unit LoRA summon, FSDP1/2 compatibility, and strip-modules support
原始 PR · 作者 qinganrice · 合并时间 2026-06-18 11:37
重构FSDP+LoRA分层召唤,降低显存并修复死锁
该 PR 值得所有深入使用 FSDP+LoRA 训练的用户精读,尤其是 `layered_summon_lora_params` 的设计思路和 wrap 策略防护逻辑。建议结合具体模型进行回归测试,验证显存与训练稳定性。对于重复 all-gather 的风险,可在升级后仍保持监控,必要时在内部 fork 中进一步优化。
参与讨论