niansa/libjustlm
1
0
Fork 0
mirror of https://gitlab.com/niansa/libjustlm.git synced 2025-03-06 20:49:17 +01:00
Code Releases Activity

Check for errors during llama evaluation properly

Browse source
This commit is contained in:
niansa 2023-05-11 18:46:41 +02:00
parent 087fe1396b
commit 59a6a8b1d1
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
justlm_llama.hpp
View file

@ -82,7 +82,7 @@ class LLaMaInference final : public Inference {
if (it + params.n_batch >= ssize_t(state->tokens.size())) break; if (it + params.n_batch >= ssize_t(state->tokens.size())) break;
// Evaluate // Evaluate
if (!llama_eval(state->ctx, state->tokens.data()+it, params.n_batch, it, params.n_threads)) { if (llama_eval(state->ctx, state->tokens.data()+it, params.n_batch, it, params.n_threads)) {
LM_COTHROW("Failed to evaluate tokens in batches", LM_BOOL_ERROR); LM_COTHROW("Failed to evaluate tokens in batches", LM_BOOL_ERROR);
} }
@ -99,7 +99,7 @@ class LLaMaInference final : public Inference {
// Evaluate remaining tokens // Evaluate remaining tokens
if (it < state->tokens.size()) { if (it < state->tokens.size()) {
for (; it != state->tokens.size(); it++) { for (; it != state->tokens.size(); it++) {
if (!llama_eval(state->ctx, state->tokens.data()+it, 1, it, params.n_threads)) { if (llama_eval(state->ctx, state->tokens.data()+it, 1, it, params.n_threads)) {
LM_COTHROW("Failed to evaluate individual tokens", LM_BOOL_ERROR); LM_COTHROW("Failed to evaluate individual tokens", LM_BOOL_ERROR);
} }
} }
@ -186,7 +186,7 @@ public:
// Evaluate token // Evaluate token
// TODO: Respect batch size // TODO: Respect batch size
if (!llama_eval(state->ctx, state->tokens.data()+state->tokens.size()-1, 1, state->tokens.size()-1, params.n_threads)) { if (llama_eval(state->ctx, state->tokens.data()+state->tokens.size()-1, 1, state->tokens.size()-1, params.n_threads)) {
LM_COTHROW("Failed to evaluate new tokens", ""); LM_COTHROW("Failed to evaluate new tokens", "");
} }