fix(checkpoint): preserve tied lm_head on resume - #2511

Merged

akoumpa merged 6 commits into

mainfrom

yuhez/fix/tied-lm-head-resume

Jun 15, 2026

Commits on Jun 10, 2026

fix(checkpoint): preserve tied lm_head on resume
yuhezhang-ai
committed
60544dd
docs(checkpoint): clarify tied lm_head storage check
yuhezhang-ai
committed
dd99ba0

Commits on Jun 11, 2026

fix(checkpoint): retie local lm_head after sharding
yuhezhang-ai
committed
426a63c
Merge remote-tracking branch 'origin/main' into yuhez/fix/tied-lm-head-resume
yuhezhang-ai
committed
dbeb5c8
fix(checkpoint): refresh tied lm head state during DCP
yuhezhang-ai
committed
50f781e

Commits on Jun 15, 2026

Merge branch 'main' into yuhez/fix/tied-lm-head-resume
akoumpa
authored
406d593