Rankings_watch displaying wrong sequence

[MiriUll]

Hello,
I have a problem with the rankings_watch() function. I used a predefined GPT2 model and gave it the input "Today, the weather is". However, in the visualization, only the first token is shown although the model creates the output correctly:

Thank you for your help :D

[jalammar]

Yep, I can replicate this. It's a an issue with the title of the figure. But the values are correct for the 5th position.

Notice how "getting" is number 1 and is what the model actually outputs.