Просмотр исходного кода

Merge pull request #4736 from ollama/mxyng/vocab-only

vocab only for tokenize
Michael Yang 11 месяцев назад
Родитель
Сommit
eb2c443a79
1 измененных файлов с 4 добавлено и 4 удалено
  1. 4 4
      llm/llm.go

+ 4 - 4
llm/llm.go

@@ -47,11 +47,11 @@ func newLlamaModel(p string) *llamaModel {
 	cs := C.CString(p)
 	defer C.free(unsafe.Pointer(cs))
 
+	params := C.llama_model_default_params()
+	params.vocab_only = true
+
 	return &llamaModel{
-		C.llama_load_model_from_file(
-			cs,
-			C.llama_model_default_params(),
-		),
+		C.llama_load_model_from_file(cs, params),
 	}
 }