diff --git a/common/sampling.cpp b/common/sampling.cpp index 8f1a7b31f..76a3bd547 100644 --- a/common/sampling.cpp +++ b/common/sampling.cpp @@ -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