Skip to content

Commit 85f9049

Browse files
abetlenDon Mahurin
authored and
Don Mahurin
committed
Fix return type
1 parent add0841 commit 85f9049

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/llama_cpp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,7 @@ def llama_get_state_size(ctx: llama_context_p) -> c_size_t:
260260
# Returns the number of bytes copied
261261
def llama_copy_state_data(
262262
ctx: llama_context_p, dest # type: Array[c_uint8]
263-
) -> c_size_t:
263+
) -> int:
264264
return _lib.llama_copy_state_data(ctx, dest)
265265

266266

@@ -272,7 +272,7 @@ def llama_copy_state_data(
272272
# Returns the number of bytes read
273273
def llama_set_state_data(
274274
ctx: llama_context_p, src # type: Array[c_uint8]
275-
) -> c_size_t:
275+
) -> int:
276276
return _lib.llama_set_state_data(ctx, src)
277277

278278

0 commit comments

Comments
 (0)