|
|
@@ -2541,8 +2541,13 @@ static void llama_sampler_infill_apply(struct llama_sampler * smpl, llama_token_
|
|
|
if (n_non_eog == 0) {
|
|
|
cur_p->size = 1;
|
|
|
cur_p->data[0].id = ctx->vocab->token_eot();
|
|
|
+ if (cur_p->data[0].id == LLAMA_TOKEN_NULL) {
|
|
|
+ cur_p->data[0].id = ctx->vocab->token_eos();
|
|
|
+ }
|
|
|
cur_p->data[0].logit = 1.0f;
|
|
|
|
|
|
+ GGML_ASSERT(cur_p->data[0].id != LLAMA_TOKEN_NULL);
|
|
|
+
|
|
|
return;
|
|
|
}
|
|
|
|