From a8551477f5c3786aaefbde4163c4a6ea0a0f4380 Mon Sep 17 00:00:00 2001 From: Vinicius Date: Thu, 20 Jul 2023 17:29:11 -0300 Subject: [PATCH] Update llama_cpp.py - Fix c_char_p to Array[c_char_p] and c_float to Array[c_float] --- 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 aef4f65c7..345d1a44f 100644 --- a/llama_cpp/llama_cpp.py +++ b/llama_cpp/llama_cpp.py @@ -748,7 +748,7 @@ def llama_get_vocab( return _lib.llama_get_vocab(ctx, strings, scores, capacity) -_lib.llama_get_vocab.argtypes = [llama_context_p, c_char_p, c_float, c_int] +_lib.llama_get_vocab.argtypes = [llama_context_p, Array[c_char_p], Array[c_float], c_int] _lib.llama_get_vocab.restype = c_int