root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-24 10:24:35 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

perplexity : fix winogrande N tasks option

Browse Source
This commit is contained in:
Georgi Gerganov 2024-01-18 20:49:00 +02:00
parent e9240cdfa0
commit d391ae9b49
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/perplexity/perplexity.cpp
View File

@ -865,7 +865,7 @@ static void winogrande_score(llama_context * ctx, const gpt_params & params) {
} }
float scale = 1/(1.f + (float)rng.max()); float scale = 1/(1.f + (float)rng.max());
std::vector<winogrande_entry> selected; std::vector<winogrande_entry> selected;
selected.reserve(params.winogrande_tasks); selected.resize(params.winogrande_tasks);
for (int i = 0; i < int(params.winogrande_tasks); ++i) { for (int i = 0; i < int(params.winogrande_tasks); ++i) {
int j = int(scale*rng()*aux.size()); int j = int(scale*rng()*aux.size());
selected[i] = std::move(data[aux[j]]); selected[i] = std::move(data[aux[j]]);