Adapt to new should_add_bos function

This commit is contained in:
Branden Butler 2023-11-18 12:23:33 -06:00
parent c2d0041704
commit e778ce4a4c

View file

@ -96,7 +96,7 @@ int main(int argc, char ** argv) {
// 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);
std::vector<llama_token> inp;