Getting probability distributions

[SeriousJ55]

LitGPT is lit! Thank you very much for your great tool and your support!

Is there a way to get the probability distribution of the next word in case of a completion?

For instance, I would input "Hello, how are" and I would get something like this:

{' you': -0.41093945503234863, ' we': -2.035161256790161, ' the': -3.56518292427063, ' your': -3.6833927631378174, 'they': -3.713589906692505, ' things': -3.9313480854034424, ' I': -4.467358589172363, ' these': -4.666958808898926, 'our': -5.107480049133301, ' all': -5.128758430480957}