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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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;
}