瀏覽代碼

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

vocab only for tokenize
Michael Yang 11 月之前
父節點
當前提交
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)
 	cs := C.CString(p)
 	defer C.free(unsafe.Pointer(cs))
 	defer C.free(unsafe.Pointer(cs))
 
 
+	params := C.llama_model_default_params()
+	params.vocab_only = true
+
 	return &llamaModel{
 	return &llamaModel{
-		C.llama_load_model_from_file(
-			cs,
-			C.llama_model_default_params(),
-		),
+		C.llama_load_model_from_file(cs, params),
 	}
 	}
 }
 }