diff options
author | Jed Fox <git@jedfox.com> | 2023-05-06 17:01:47 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-06 17:01:47 -0400 |
commit | 3924088512d9e12e90ed6dbf28a6c5712481d33e (patch) | |
tree | 9dacd76924e57e792ab834b47d77b875d1c8ae4f /examples/main | |
parent | 173d0e6419e8f8f3c1f4f13201b777f4c60629f3 (diff) |
Remove default arguments from sampling functions (#1343)
Diffstat (limited to 'examples/main')
-rw-r--r-- | examples/main/main.cpp | 8 |
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); } |