diff options
Diffstat (limited to 'examples/llava/llava-cli.cpp')
| -rw-r--r-- | examples/llava/llava-cli.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/llava/llava-cli.cpp b/examples/llava/llava-cli.cpp index 75948806..50dac4ca 100644 --- a/examples/llava/llava-cli.cpp +++ b/examples/llava/llava-cli.cpp @@ -45,7 +45,7 @@ static const char * sample(struct llama_sampling_context * ctx_sampling, const llama_token id = llama_sampling_sample(ctx_sampling, ctx_llama, NULL); llama_sampling_accept(ctx_sampling, ctx_llama, id, true); static std::string ret; - if (id == llama_token_eos(llama_get_model(ctx_llama))) { + if (llama_token_is_eog(llama_get_model(ctx_llama), id)) { ret = "</s>"; } else { ret = llama_token_to_piece(ctx_llama, id); |