diff --git a/examples/llama_cpp.py b/examples/llama_cpp.py index a56243dc9..f2366effe 100644 --- a/examples/llama_cpp.py +++ b/examples/llama_cpp.py @@ -337,7 +337,7 @@ def llama_tokenize( tokens, # type: Array[llama_token] n_max_tokens: c_int, add_bos: c_bool, -) -> c_int: +) -> int: return _lib.llama_tokenize(ctx, text, tokens, n_max_tokens, add_bos)