Merge pull request abetlen#511 from viniciusarruda/patch-1

abetlen · web-flow · commit a9cb645495f3 · 2023-07-20T17:40:39.000-04:00
Update llama_cpp.py - Fix c_char_p to Array[c_char_p] and c_float to …
diff --git 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
 
 
