Fix ZeRO-3 crash in AutoTP universal-checkpoint metadata (#8270)