Skip to content

Commit

Permalink
compile fix
Browse files Browse the repository at this point in the history
  • Loading branch information
LostRuins committed Feb 4, 2024
1 parent 6dc0129 commit 38863a3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions common/sampling.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -249,10 +249,10 @@ static llama_token llama_sampling_sample_impl(
} else {
if (mirostat == 1) {
const int mirostat_m = 100;
llama_sample_temp(ctx_main, &cur_p, temp);
llama_sample_temp(ctx_main, &cur_p, temp, 0);
id = llama_sample_token_mirostat(ctx_main, &cur_p, mirostat_tau, mirostat_eta, mirostat_m, &ctx_sampling->mirostat_mu);
} else if (mirostat == 2) {
llama_sample_temp(ctx_main, &cur_p, temp);
llama_sample_temp(ctx_main, &cur_p, temp, 0);
id = llama_sample_token_mirostat_v2(ctx_main, &cur_p, mirostat_tau, mirostat_eta, &ctx_sampling->mirostat_mu);
} else {
// temperature sampling
Expand Down

0 comments on commit 38863a3

Please sign in to comment.