Skip to content

Commit 29e8c67

Browse files
authored
fix: adapt to llama.cpp changes (#208)
1 parent 6b012a6 commit 29e8c67

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llama/addon.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,10 +167,10 @@ class LLAMAContext : public Napi::ObjectWrap<LLAMAContext> {
167167
}
168168

169169
ctx = llama_new_context_with_model(model->model, context_params);
170-
Napi::MemoryManagement::AdjustExternalMemory(Env(), llama_get_state_size(ctx));
170+
Napi::MemoryManagement::AdjustExternalMemory(Env(), llama_state_get_size(ctx));
171171
}
172172
~LLAMAContext() {
173-
Napi::MemoryManagement::AdjustExternalMemory(Env(), -(int64_t)llama_get_state_size(ctx));
173+
Napi::MemoryManagement::AdjustExternalMemory(Env(), -(int64_t)llama_state_get_size(ctx));
174174
llama_free(ctx);
175175
model->Unref();
176176
}

0 commit comments

Comments
 (0)