djl icon indicating copy to clipboard operation
djl copied to clipboard

Expose vLLM logprobs in model output

Open CoolFish88 opened this issue 1 year ago • 3 comments
trafficstars

Description

vLLM sampling parameters include a richer set of values, among which logprobs has a wider utility.

When testing by adding the logpobs option to the request payload, the model output schema was unchanged {"generated text": "model_output"} suggesting it was not propagated to the output

Will this change the current api? How?

Probably by enriching the output schema.

Who will benefit from this enhancement?

Anyone who wants logprobs extracted from model predictions.

References

  • list known implementations This thread provides a starting point for tackling this issue.

CoolFish88 avatar Oct 01 '24 20:10 CoolFish88