vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix type signature of token_to_str

Browse source
This commit is contained in:
Andrei Betlen 2023-03-31 03:25:12 -04:00 • committed by Don Mahurin
parent a7a6d88793
commit 5bb1bc74d1
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/llama_cpp.py
View file

@ -182,7 +182,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)