diff --git a/examples/main/main.cpp b/examples/main/main.cpp index 9b49eaeb4..b78a88b06 100644 --- a/examples/main/main.cpp +++ b/examples/main/main.cpp @@ -430,12 +430,12 @@ int main(int argc, char ** argv) { last_n_tokens.data() + last_n_tokens.size() - last_n_repeat, last_n_repeat, alpha_frequency, alpha_presence); if (!penalize_nl) { - for (size_t idx = 0; idx < candidates_p.size; idx++) { - if (candidates_p.data[idx].id == llama_token_nl()) { - candidates_p.data[idx].logit = nl_logit; - break; + for (size_t idx = 0; idx < candidates_p.size; idx++) { + if (candidates_p.data[idx].id == llama_token_nl()) { + candidates_p.data[idx].logit = nl_logit; + break; + } } - } } if (temp <= 0) {