Skip to content

Commit 1410813

Browse files
committed
Fix pushing in wrong halflayer idx
1 parent 74eebc6 commit 1410813

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/perplexity/perplexity.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -405,7 +405,7 @@ static results_perplexity perplexity(llama_context * ctx, const gpt_params & par
405405
std::get<1>(pass_results[temp]), std::get<2>(pass_results[temp]));
406406
if (anti_mode) {
407407
skip_types[lidx] |= std::get<1>(pass_results[temp]);
408-
skips.push_back(std::get<1>(pass_results[temp]) == 1 ? lidx : -lidx);
408+
skips.push_back(std::get<1>(pass_results[temp]) == 1 ? lidx : lidx + n_layers);
409409
}
410410
if (++pruned >= num_prune) break;
411411
}

0 commit comments

Comments
 (0)