This commit is contained in:
jmorganca 2024-08-13 11:24:46 -07:00
parent b267ab92b0
commit a483a4c4ed
2 changed files with 2 additions and 1 deletions

View File

@ -44,6 +44,7 @@ package llama
bool llamaProgressCallback(float progress, void *user_data); bool llamaProgressCallback(float progress, void *user_data);
*/ */
import "C" import "C"
import ( import (
_ "embed" _ "embed"
"errors" "errors"
@ -405,7 +406,6 @@ func (s *SamplingContext) Sample(ctxMain *Context, ctxConfig *Context, idx int)
} }
return int(C.llama_sampling_csample(s.c, ctxMain.c, ctxConfig.c, C.int(idx))) return int(C.llama_sampling_csample(s.c, ctxMain.c, ctxConfig.c, C.int(idx)))
} }
func (s *SamplingContext) Accept(ctxMain *Context, id int, applyGrammar bool) { func (s *SamplingContext) Accept(ctxMain *Context, id int, applyGrammar bool) {

View File

@ -3,6 +3,7 @@ package llama
// extern const char *ggml_metallib_start; // extern const char *ggml_metallib_start;
// extern const char *ggml_metallib_end; // extern const char *ggml_metallib_end;
import "C" import "C"
import ( import (
_ "embed" _ "embed"
"strings" "strings"