Skip to content

Commit 471e469

Browse files
committed
pre gguf merge
1 parent d50ccb0 commit 471e469

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/perplexity/perplexity.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ std::vector<float> hellaswag_evaluate_tokens(llama_context * ctx, const std::vec
130130
for (size_t i_chunk = 0; i_chunk < n_chunk; ++i_chunk) {
131131
size_t n_tokens = tokens.size() - i_chunk * n_batch;
132132
n_tokens = std::min(n_tokens, size_t(n_batch));
133-
if (llama_eval(ctx, tokens.data() + i_chunk * n_batch, n_tokens, n_past, n_thread)) {
133+
if (llama_eval(ctx, tokens.data() + i_chunk * n_batch, n_tokens, n_past, n_thread, n_thread)) {
134134
fprintf(stderr, "%s : failed to eval\n", __func__);
135135
return {};
136136
}
@@ -304,7 +304,7 @@ void hellaswag_score(llama_context * ctx, const gpt_params & params) {
304304
//}
305305

306306
// Evaluate the query
307-
logits = hellaswag_evaluate_tokens(ctx, query_embd, context_size, params.n_batch, n_vocab, params.n_threads, params.pp_threads);
307+
logits = hellaswag_evaluate_tokens(ctx, query_embd, context_size, params.n_batch, n_vocab, params.n_threads);
308308
if (logits.empty()) {
309309
fprintf(stderr, "%s : failed to eval\n", __func__);
310310
return;

0 commit comments

Comments
 (0)