size_t

0142660b · Michael Yang · 743e957d · 0142660b
Commit 0142660b authored Jul 14, 2023 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

llama/llama.go llama/llama.go +2 -2

No files found.
--- a/llama/llama.go
+++ b/llama/llama.go
@@ -264,8 +264,8 @@ func (llm *llama) sample(output deque[C.llama_token], opts *C.struct_llama_sampl

 	token := C.llama_sample(
 		llm.ctx,
-		unsafe.SliceData(candidates.Data()), C.ulong(candidates.Len()),
-		unsafe.SliceData(output.Data()), C.ulong(output.Len()),
+		unsafe.SliceData(candidates.Data()), C.size_t(candidates.Len()),
+		unsafe.SliceData(output.Data()), C.size_t(output.Len()),
 		opts)
 	if token != C.llama_token_eos() {
 		return token, nil