text-generation-inference icon indicating copy to clipboard operation
text-generation-inference copied to clipboard

[Feature] Add contrastive sampling

Open yk opened this issue 2 years ago • 3 comments

HF has contrastive sampling, ideally we could use that to sample here too.

yk avatar Mar 08 '23 22:03 yk

Contrastive sampling is very expensive to compute and would require a large change in the codebase to be supported. If there is enough interest we can add it but for now it is not a high priority.

OlivierDehaene avatar Mar 09 '23 08:03 OlivierDehaene

Well at the same time I added best_of which is even more expensive...

OlivierDehaene avatar Mar 10 '23 09:03 OlivierDehaene

haha, no worries :) we saw some good results with contrastive. but not urgent at all. thanks for the consideration

yk avatar Mar 10 '23 22:03 yk