Adapt to new should_add_bos function
This commit is contained in:
parent
c2d0041704
commit
e778ce4a4c
1 changed files with 1 additions and 1 deletions
|
@ -96,7 +96,7 @@ int main(int argc, char ** argv) {
|
||||||
|
|
||||||
|
|
||||||
// Tokenize the prompt
|
// Tokenize the prompt
|
||||||
const bool add_bos = llama_vocab_type(llama_get_model(ctx_tgt)) == LLAMA_VOCAB_TYPE_SPM;
|
const bool add_bos = llama_should_add_bos_token(model_tgt);
|
||||||
LOG("add_bos: %d\n", add_bos);
|
LOG("add_bos: %d\n", add_bos);
|
||||||
|
|
||||||
std::vector<llama_token> inp;
|
std::vector<llama_token> inp;
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue