abetlen / abetlen/llama-cpp-python

Support for a limited vocabulary for generation

未關閉
#998 7 則留言 0 個 reaction 已指派 0 人 在 GitHub 檢視
question
主要語言
Python
星號
10.6k
分支
1.4k
PR 合併指標
PR 指標待擷取

描述

**Is your feature request related to a problem? Please describe.**
I would like to constrain the model output to only use a custom vocabulary comprising a list of allowable words (or alternatively, to blacklist all other words in the vocabulary).

**Describe the solution you'd like**
HuggingFace's transformer library features a `bad_words_id` keyword in the `model.generate` function that accepts a list of words to exclude from its output (some discussion of this feature [here](https://github.com/huggingface/transformers/issues/21961)).

**Describe alternatives you've considered**
Could this possibly be achieved with the use of a `llama_cpp.LogitsProcessor`? I am less familiar with this library and haven't found examples in a similar direction, so am unsure how straightforward this could be to implement using one of those.

貢獻指南

開啟貢獻指南

評估

這個 Issue 還沒有評估資料。

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。