Skip to content

Commit c6123e6

Browse files
committed
added top-k sampler to improve performance
1 parent 6664d47 commit c6123e6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

common/sampling.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,9 @@ struct common_sampler * common_sampler_init(const struct llama_model * model, co
168168

169169
if (params.mirostat == 0) {
170170
if(params.top_n_sigma >= 0) {
171-
llama_sampler_chain_add(result->chain, llama_sampler_init_temp(params.temp));
172-
llama_sampler_chain_add(result->chain, llama_sampler_init_top_n_sigma(params.top_n_sigma));
171+
llama_sampler_chain_add(result->chain, llama_sampler_init_top_k (params.top_k));
172+
llama_sampler_chain_add(result->chain, llama_sampler_init_temp (params.temp));
173+
llama_sampler_chain_add(result->chain, llama_sampler_init_top_n_sigma (params.top_n_sigma));
173174
} else {
174175
for (const auto & cnstr : params.samplers) {
175176
switch (cnstr) {

0 commit comments

Comments
 (0)