But the probability vector is the output of the LLM, no?