IBM / text-generation-inference

IBM development fork of https://github.com/huggingface/text-generation-inference
Apache License 2.0
52 stars 30 forks source link

TGIS gRPC adapter for lm-eval #90

Open maxdebayser opened 4 months ago

maxdebayser commented 4 months ago

(Moving this PR over here)

This PoC adds a backend to lm-eval so that it can call a running TGIS or tgis-vllm server over grpc. It can run benchmarks based on the generate function for decoder and encoder-decoder models. For the logprobs function only decoder models are supported because tgis doesn't return the input logprobs for encoder-decoder models.