Fix type signature of token_to_str
This commit is contained in:
parent
a7a6d88793
commit
5bb1bc74d1
1 changed files with 1 additions and 1 deletions
|
@ -182,7 +182,7 @@ _lib.llama_get_embeddings.restype = POINTER(c_float)
|
||||||
|
|
||||||
|
|
||||||
# Token Id -> String. Uses the vocabulary in the provided context
|
# 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)
|
return _lib.llama_token_to_str(ctx, token)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue