Skip to content

Commit 8dfde63

Browse files
committed
Fix return type
1 parent 2753b85 commit 8dfde63

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llama_cpp/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)