lightllm icon indicating copy to clipboard operation
lightllm copied to clipboard

What is the plan to support beam search

Open feifeibear opened this issue 1 year ago • 4 comments

I have noticed that LightLLM currently seems to only support decoding through sampling. Additional decoding methods such as BeamSearch and GreedySearch are not yet supported. I would like to know about the development plans in this regard.

feifeibear avatar Jan 08 '24 02:01 feifeibear

We have an internal version that supports beam search, but we feel like beam search doesn't add much value to diversity generation, so we have been keeping it out to maintain simplicity in the framework. Is there a rigid use case where beam search is a must-have?

hiworldwzj avatar Jan 08 '24 06:01 hiworldwzj

I believe that beam search is a must-have feature for the LLM serving framework.

feifeibear avatar Jan 09 '24 02:01 feifeibear

@feifeibear ok, we will support this feature soon.

hiworldwzj avatar Jan 09 '24 02:01 hiworldwzj

@feifeibear ok, we will support this feature soon.

Thanks! let me if you need help.

feifeibear avatar Jan 09 '24 05:01 feifeibear