diff options
Diffstat (limited to 'examples/llava/llava-utils.h')
| -rw-r--r-- | examples/llava/llava-utils.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/llava/llava-utils.h b/examples/llava/llava-utils.h index 45b2b1ad..320c7196 100644 --- a/examples/llava/llava-utils.h +++ b/examples/llava/llava-utils.h @@ -137,7 +137,7 @@ inline llama_token sample_id(llama_context * ctx_llama, gpt_params & params) { inline const char * sample(struct llama_context * ctx_llama, gpt_params & params, int * n_past) { int id = sample_id(ctx_llama, params); static std::string ret; - if (id == llama_token_eos(ctx_llama)) { + if (id == llama_token_eos(llama_get_model(ctx_llama))) { ret = "</s>"; } else { ret = llama_token_to_piece(ctx_llama, id); |