Include token probabilities in TopTokensScore