python analyze.py plot-token-probs