Fix ZeroDivisionError in SFT during record_train_metrics#3581

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

igorts/sft-timing-fixAI-Hypercomputer/maxtext:igorts/sft-timing-fixCopy head branch name to clipboard