vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update utils.cpp

Browse source
This commit is contained in:
Georgi Gerganov 2023-03-12 22:22:37 +02:00 • committed by GitHub
parent 7412189fe5
commit 2913d75a42
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
utils.cpp
View file

@ -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 // find the top K tokens
std::partial_sort( std::partial_sort(
logits_id.begin(), logits_id.begin(),