alihan/llama-cpp-python
1
0
Fork 0
mirror of https://github.com/abetlen/llama-cpp-python.git synced 2023-09-07 17:34:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix type signature of token_to_str

Browse Source
This commit is contained in:
Andrei Betlen
2023-03-31 03:25:12 -04:00
parent 670d390001
commit 49c8df369a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
llama_cpp/llama_cpp.py
View File

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