From 38f442deb03974261a4495d93ee9d08f08126dac Mon Sep 17 00:00:00 2001 From: Andrei Betlen Date: Sat, 8 Apr 2023 15:05:33 -0400 Subject: [PATCH] Bugfix: Wrong size of embeddings. Closes #47 --- llama_cpp/llama_cpp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_cpp/llama_cpp.py b/llama_cpp/llama_cpp.py index 2d4646f..069ae10 100644 --- a/llama_cpp/llama_cpp.py +++ b/llama_cpp/llama_cpp.py @@ -218,7 +218,7 @@ _lib.llama_n_ctx.restype = c_int def llama_n_embd(ctx: llama_context_p) -> c_int: - return _lib.llama_n_ctx(ctx) + return _lib.llama_n_embd(ctx) _lib.llama_n_embd.argtypes = [llama_context_p]