Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 0142660b authored by Michael Yang's avatar Michael Yang
Browse files

size_t

parent 743e957d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
llama/llama.go
View file @ 0142660b
...@@ -264,8 +264,8 @@ func (llm *llama) sample(output deque[C.llama_token], opts *C.struct_llama_sampl ...@@ -264,8 +264,8 @@ func (llm *llama) sample(output deque[C.llama_token], opts *C.struct_llama_sampl
token := C.llama_sample( token := C.llama_sample(
llm.ctx, llm.ctx,
unsafe.SliceData(candidates.Data()), C.ulong(candidates.Len()), unsafe.SliceData(candidates.Data()), C.size_t(candidates.Len()),
unsafe.SliceData(output.Data()), C.ulong(output.Len()), unsafe.SliceData(output.Data()), C.size_t(output.Len()),
opts) opts)
if token != C.llama_token_eos() { if token != C.llama_token_eos() {
return token, nil return token, nil
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment