microsoft / DeepSpeed-MII

MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.
Apache License 2.0
1.85k stars 174 forks source link

Compute perplexity #496

Open Sh1gechan opened 3 months ago

Sh1gechan commented 3 months ago

Is there any way to retrieve logprob in DeepSpeed-mii, just as we retrieve logprob by specifying options in vLLM? My intention is to find out how the output from deepspeed-mii, vLLM and huggingface pipeline differs by retrieving the probability distribution.

Related Issues