From 49c8df369a61d32652d7bcbd958debdbf08a1fea Mon Sep 17 00:00:00 2001 From: Andrei Betlen Date: Fri, 31 Mar 2023 03:25:12 -0400 Subject: [PATCH] Fix type signature of token_to_str --- 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 fdad289..a6da424 100644 --- a/llama_cpp/llama_cpp.py +++ b/llama_cpp/llama_cpp.py @@ -187,7 +187,7 @@ _lib.llama_get_embeddings.restype = POINTER(c_float) # Token Id -> String. Uses the vocabulary in the provided context -def llama_token_to_str(ctx: llama_context_p, token: int) -> bytes: +def llama_token_to_str(ctx: llama_context_p, token: llama_token) -> bytes: return _lib.llama_token_to_str(ctx, token)