From 3bf60c5eb30d77d64f553f1ab507f277f7ed7dda Mon Sep 17 00:00:00 2001 From: Evan Jones Date: Wed, 23 Aug 2023 20:49:47 -0400 Subject: [PATCH] llama : fix grammar sometimes generating null char --- llama.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama.cpp b/llama.cpp index f2dc4da1d..7cac8a1ce 100644 --- a/llama.cpp +++ b/llama.cpp @@ -4074,7 +4074,7 @@ void llama_sample_grammar(struct llama_context * ctx, llama_token_data_array * c if (!allow_eos) { candidates->data[i].logit = -INFINITY; } - } else if (text.empty()) { + } else if (text.empty() || text[0] == 0) { candidates->data[i].logit = -INFINITY; } else { candidates_decoded.push_back(decode_utf8(text.c_str(), grammar->partial_utf8));