llama : fix grammar sometimes generating null char (#2756)

This commit is contained in:
Evan Jones 2023-08-24 00:07:13 -04:00 committed by GitHub
parent 44d5462b5c
commit 6e91a1b070
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -4074,7 +4074,7 @@ void llama_sample_grammar(struct llama_context * ctx, llama_token_data_array * c
if (!allow_eos) { if (!allow_eos) {
candidates->data[i].logit = -INFINITY; candidates->data[i].logit = -INFINITY;
} }
} else if (text.empty()) { } else if (text.empty() || text[0] == 0) {
candidates->data[i].logit = -INFINITY; candidates->data[i].logit = -INFINITY;
} else { } else {
candidates_decoded.push_back(decode_utf8(text.c_str(), grammar->partial_utf8)); candidates_decoded.push_back(decode_utf8(text.c_str(), grammar->partial_utf8));