early stop when all sequence reach EOS#57

je1lee wants to merge 4 commits intogoogle:maingoogle/gemma_pytorch:mainfrom

je1lee:fix/earlystopje1lee/gemma_pytorch:fix/earlystopCopy head branch name to clipboard