Fix distillation checkpoint on resumption#3464

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

agagik-distill-checkpointAI-Hypercomputer/maxtext:agagik-distill-checkpointCopy head branch name to clipboard