autolabel icon indicating copy to clipboard operation
autolabel copied to clipboard

[Feature Request]: Support logit_bias to constrain LLM outputs

Open nihit opened this issue 2 years ago • 3 comments

Is your feature request related to a problem? Please describe. LLM might sometimes have trouble following output format instructions

Describe the solution you'd like logit_bias: https://twitter.com/AAAzzam/status/1669753725093654565

nihit avatar Jun 21 '23 01:06 nihit

We can start with support for classification and entity matching tasks

nihit avatar Jun 21 '23 01:06 nihit

thanks @neildhruva for the suggestion!

nihit avatar Jun 21 '23 01:06 nihit

logit_bias is an OpenAI only feature at the moment. For huggingface equivalent, we can use this: https://huggingface.co/docs/transformers/main/en/internal/generation_utils#transformers.SequenceBiasLogitsProcessor

DhruvaBansal00 avatar Jun 22 '23 17:06 DhruvaBansal00