sampling : make top_n_sigma no-op at <=0 or a single candidate (#13345)

This commit is contained in:
DocShotgun
2025-05-06 13:36:24 -07:00
committed by GitHub
parent 91a86a6f35
commit ffc727203a
2 changed files with 2 additions and 2 deletions

View File

@ -1750,7 +1750,7 @@ static const char * llama_sampler_top_n_sigma_name(const struct llama_sampler *
static void llama_sampler_top_n_sigma_apply(struct llama_sampler * smpl, llama_token_data_array * cur_p) {
const auto * ctx = (llama_sampler_top_n_sigma *) smpl->ctx;
if (ctx->n < 0.0f) {
if (ctx->n <= 0.0f || cur_p->size <= 1) {
return;
}