summaryrefslogtreecommitdiff
path: root/common/sampling.cpp
diff options
context:
space:
mode:
authorRobey Holderith <robey@flaminglunchbox.net>2024-02-18 11:11:16 -0800
committerGitHub <noreply@github.com>2024-02-18 21:11:16 +0200
commit5ee99c32f5e47c8d32634eff9a47fb32a24c276b (patch)
treeb96e45b36fe325a926025f94655a885428c84f47 /common/sampling.cpp
parentc145f8a132b2fe1d1e65987faddbd9a40bef7a12 (diff)
common, server : surface min_keep as its own parameter (#5567)
* Feature - surface min_keep as its own parameter * Updated README with min_keep param
Diffstat (limited to 'common/sampling.cpp')
-rw-r--r--common/sampling.cpp5
1 files changed, 4 insertions, 1 deletions
diff --git a/common/sampling.cpp b/common/sampling.cpp
index 611c327b..de4331a1 100644
--- a/common/sampling.cpp
+++ b/common/sampling.cpp
@@ -248,7 +248,10 @@ static llama_token llama_sampling_sample_impl(
llama_sample_temp(ctx_main, &cur_p, temp);
id = llama_sample_token_mirostat_v2(ctx_main, &cur_p, mirostat_tau, mirostat_eta, &ctx_sampling->mirostat_mu);
} else {
- sampler_queue(ctx_main, params, cur_p, 1);
+ // temperature sampling
+ size_t min_keep = std::max(1, params.min_keep);
+
+ sampler_queue(ctx_main, params, cur_p, min_keep);
id = llama_sample_token(ctx_main, &cur_p);