llama : define non-positive top_k; top_k range check (#779)

* Define non-positive top_k; top_k range check

* minor : brackets

---------

Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
pull/783/head master-5a8c4f6
Ivan Stepanov 1 year ago committed by GitHub
parent ff05d05c96
commit 5a8c4f6240
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1229,7 +1229,9 @@ static llama_vocab::id llama_sample_top_p_top_k(
}
}
sample_top_k(logits_id, top_k);
if (top_k > 0 && top_k < n_logits) {
sample_top_k(logits_id, top_k);
}
float maxl = -std::numeric_limits<float>::infinity();
for (const auto & kv : logits_id) {

Loading…
Cancel
Save