Update utils.cpp

pull/56/head
Georgi Gerganov 1 year ago committed by GitHub
parent 7412189fe5
commit 2913d75a42
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -296,7 +296,7 @@ bool gpt_vocab_init(const std::string & fname, gpt_vocab & vocab) {
}
void sample_top_k(std::vector<std::pair<double, gpt_vocab::id>> &logits_id, int top_k) {
void sample_top_k(std::vector<std::pair<double, gpt_vocab::id>> & logits_id, int top_k) {
// find the top K tokens
std::partial_sort(
logits_id.begin(),

Loading…
Cancel
Save