Skip to content

Commit efce1f4

Browse files
committed
llama_sampler_init params fixed
1 parent 6cb058a commit efce1f4

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

llama_cpp/llama_cpp.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3419,7 +3419,7 @@ class llama_sampler(ctypes.Structure):
34193419
("clone", llama_sampler_i_clone),
34203420
("free", llama_sampler_i_free),
34213421
]
3422-
3422+
llama_sampler_i_p = CtypesPointer[llama_sampler_i]
34233423

34243424
# // mirror of llama_sampler_i:
34253425

@@ -3429,7 +3429,10 @@ class llama_sampler(ctypes.Structure):
34293429
[ctypes.POINTER(llama_sampler_i), llama_sampler_context_t],
34303430
llama_sampler_p_ctypes,
34313431
)
3432-
def llama_sampler_init(smpl: llama_sampler_p, /) -> llama_sampler_p:
3432+
def llama_sampler_init(
3433+
iface:llama_sampler_i_p,
3434+
ctx :llama_sampler_context_t
3435+
) -> llama_sampler_p:
34333436
...
34343437

34353438
# LLAMA_API const char * llama_sampler_name (const struct llama_sampler * smpl);

0 commit comments

Comments
 (0)