From 0e6cbff1b7509628c588e661166f6e187137734d Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Tue, 2 May 2023 23:09:08 +0300 Subject: llama : fix compile warnings --- tests/test-sampling.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tests') diff --git a/tests/test-sampling.cpp b/tests/test-sampling.cpp index 7eee4f6..8ce59af 100644 --- a/tests/test-sampling.cpp +++ b/tests/test-sampling.cpp @@ -131,7 +131,7 @@ void test_repetition_penalty( llama_token_data_array candidates_p = { candidates.data(), candidates.size(), false }; llama_sample_softmax(nullptr, &candidates_p); DUMP(&candidates_p); - llama_sample_repetition_penalty(nullptr, &candidates_p, (llama_token *)last_tokens.data(), last_tokens.size(), penalty); + llama_sample_repetition_penalty(nullptr, &candidates_p, (const llama_token *) last_tokens.data(), last_tokens.size(), penalty); llama_sample_softmax(nullptr, &candidates_p); DUMP(&candidates_p); @@ -160,7 +160,7 @@ void test_frequency_presence_penalty( llama_token_data_array candidates_p = { candidates.data(), candidates.size(), false }; llama_sample_softmax(nullptr, &candidates_p); // DUMP(&candidates_p); - llama_sample_frequency_and_presence_penalties(nullptr, &candidates_p, (llama_token *)last_tokens.data(), last_tokens.size(), alpha_frequency, alpha_presence); + llama_sample_frequency_and_presence_penalties(nullptr, &candidates_p, (const llama_token *) last_tokens.data(), last_tokens.size(), alpha_frequency, alpha_presence); llama_sample_softmax(nullptr, &candidates_p); // DUMP(&candidates_p); -- cgit v1.2.3