diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2023-04-29 09:51:06 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-29 09:51:06 +0300 |
commit | 334637e43e3a0529b4b50e2c22968b1ed1633353 (patch) | |
tree | 7617fb1dbcd0a42d45e959dfb9f4857156af5e31 /examples/main | |
parent | dd7eff57d8491792010b1002b8de6a4b54912e5c (diff) |
common : change default parameters to pre-#1126 (#1223)
Diffstat (limited to 'examples/main')
-rw-r--r-- | examples/main/main.cpp | 24 |
1 files changed, 12 insertions, 12 deletions
diff --git a/examples/main/main.cpp b/examples/main/main.cpp index 674920b..990d0fa 100644 --- a/examples/main/main.cpp +++ b/examples/main/main.cpp @@ -387,19 +387,19 @@ int main(int argc, char ** argv) { if ((int) embd_inp.size() <= n_consumed && !is_interacting) { // out of user input, sample next token - const float temp = params.temp; - const int32_t top_k = params.top_k <= 0 ? llama_n_vocab(ctx) : params.top_k; - const float top_p = params.top_p; - const float tfs_z = params.tfs_z; - const float typical_p = params.typical_p; - const int32_t repeat_last_n = params.repeat_last_n < 0 ? n_ctx : params.repeat_last_n; - const float repeat_penalty = params.repeat_penalty; - const float alpha_presence = params.presence_penalty; + const float temp = params.temp; + const int32_t top_k = params.top_k <= 0 ? llama_n_vocab(ctx) : params.top_k; + const float top_p = params.top_p; + const float tfs_z = params.tfs_z; + const float typical_p = params.typical_p; + const int32_t repeat_last_n = params.repeat_last_n < 0 ? n_ctx : params.repeat_last_n; + const float repeat_penalty = params.repeat_penalty; + const float alpha_presence = params.presence_penalty; const float alpha_frequency = params.frequency_penalty; - const int mirostat = params.mirostat; - const float mirostat_tau = params.mirostat_tau; - const float mirostat_eta = params.mirostat_eta; - const bool penalize_nl = params.penalize_nl; + const int mirostat = params.mirostat; + const float mirostat_tau = params.mirostat_tau; + const float mirostat_eta = params.mirostat_eta; + const bool penalize_nl = params.penalize_nl; // optionally save the session on first sample (for faster prompt loading next time) if (!path_session.empty() && need_to_save_session) { |