llama : fix empty ring buffer push (#9358)

This commit is contained in:
Georgi Gerganov
2024-09-08 00:33:33 +03:00
committed by GitHub
parent faf69d4237
commit f12295b8a9
2 changed files with 4 additions and 2 deletions

View File

@@ -1226,7 +1226,9 @@ static struct llama_sampler_i llama_sampler_penalties_i = {
/* .name = */ [](const struct llama_sampler * /*smpl*/) { return "penalties"; },
/* .accept = */ [](struct llama_sampler * smpl, llama_token token) {
auto * ctx = (llama_sampler_penalties *) smpl->ctx;
ctx->prev.push_back(token);
if (ctx->prev.size()) {
ctx->prev.push_back(token);
}
},
/* .apply = */ [](struct llama_sampler * smpl, llama_token_data_array * cur_p) {
auto * ctx = (llama_sampler_penalties *) smpl->ctx;