Hi, we'd like to have log probabilities for tokens returned from the model, in addition to the token ids. Can you help with this feature request?