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

Fix return type

Browse source
This commit is contained in:
Andrei Betlen 2023-05-07 19:30:14 -04:00
parent 2753b85321
commit 8dfde63255
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
llama_cpp/llama_cpp.py
View file

@ -260,7 +260,7 @@ _lib.llama_get_state_size.restype = c_size_t
# Returns the number of bytes copied
def llama_copy_state_data(
ctx: llama_context_p, dest # type: Array[c_uint8]
) -> c_size_t:
) -> int:
return _lib.llama_copy_state_data(ctx, dest)
@ -272,7 +272,7 @@ _lib.llama_copy_state_data.restype = c_size_t
# Returns the number of bytes read
def llama_set_state_data(
ctx: llama_context_p, src # type: Array[c_uint8]
) -> c_size_t:
) -> int:
return _lib.llama_set_state_data(ctx, src)