Skip to content

Commit 58828c2

Browse files
Merge pull request #8 from WangHaoranRobin/robin_fork_master
server: fix llama_sample_top_k order
2 parents c7f7f13 + bc88fec commit 58828c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/server/server.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,10 +378,10 @@ struct llama_server_context {
378378
} else {
379379
// Temperature sampling
380380
size_t min_keep = std::max(1, n_probs);
381+
llama_sample_top_k(ctx, &candidates_p, top_k, min_keep);
381382
llama_sample_tail_free(ctx, &candidates_p, tfs_z, min_keep);
382383
llama_sample_typical(ctx, &candidates_p, typical_p, min_keep);
383384
llama_sample_top_p(ctx, &candidates_p, top_p, min_keep);
384-
llama_sample_top_k(ctx, &candidates_p, top_k, min_keep);
385385
llama_sample_temperature(ctx, &candidates_p, temp);
386386
result.tok = llama_sample_token(ctx, &candidates_p);
387387
}

0 commit comments

Comments
 (0)