fix: forward padding_mask through MTP recompute path#4983

ko3n1g wants to merge 1 commit into

NVIDIA:mainNVIDIA/Megatron-LM:mainfrom

ko3n1g:ko3n1g/fix/mtp-checkpointed-forward-padding-maskko3n1g/Megatron-LM:ko3n1g/fix/mtp-checkpointed-forward-padding-maskCopy head branch name to clipboard