aboutsummaryrefslogtreecommitdiff
path: root/examples/main
diff options
context:
space:
mode:
authorJed Fox <git@jedfox.com>2023-05-06 17:01:47 -0400
committerGitHub <noreply@github.com>2023-05-06 17:01:47 -0400
commit3924088512d9e12e90ed6dbf28a6c5712481d33e (patch)
tree9dacd76924e57e792ab834b47d77b875d1c8ae4f /examples/main
parent173d0e6419e8f8f3c1f4f13201b777f4c60629f3 (diff)
Remove default arguments from sampling functions (#1343)
Diffstat (limited to 'examples/main')
-rw-r--r--examples/main/main.cpp8
1 files changed, 4 insertions, 4 deletions
diff --git a/examples/main/main.cpp b/examples/main/main.cpp
index 43dca8e..5ac151e 100644
--- a/examples/main/main.cpp
+++ b/examples/main/main.cpp
@@ -444,10 +444,10 @@ int main(int argc, char ** argv) {
id = llama_sample_token_mirostat_v2(ctx, &candidates_p, mirostat_tau, mirostat_eta, &mirostat_mu);
} else {
// Temperature sampling
- llama_sample_top_k(ctx, &candidates_p, top_k);
- llama_sample_tail_free(ctx, &candidates_p, tfs_z);
- llama_sample_typical(ctx, &candidates_p, typical_p);
- llama_sample_top_p(ctx, &candidates_p, top_p);
+ llama_sample_top_k(ctx, &candidates_p, top_k, 1);
+ llama_sample_tail_free(ctx, &candidates_p, tfs_z, 1);
+ llama_sample_typical(ctx, &candidates_p, typical_p, 1);
+ llama_sample_top_p(ctx, &candidates_p, top_p, 1);
llama_sample_temperature(ctx, &candidates_p, temp);
id = llama_sample_token(ctx, &candidates_p);
}