Fixes for llama 3.1 training#823

sean-smith wants to merge 1 commit intomlcommons:mastermlcommons/training:masterfrom

sean-smith:mastersean-smith/training:masterCopy head branch name to clipboard