compile fix

This commit is contained in:
Concedo 2024-02-04 20:30:06 +08:00
parent 6dc01297f8
commit 38863a34d2

View file

@ -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