Bugfix: Wrong size of embeddings. Closes #47

This commit is contained in:
Andrei Betlen 2023-04-08 15:05:33 -04:00
parent 6d1bda443e
commit 38f442deb0

View file

@ -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]