llama : fix empty ring buffer push (#9358)

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

No files matched your search

+3 -1
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;