Michael Yang 11 bulan lalu
induk
melakukan
bf54c845e9
1 mengubah file dengan 4 tambahan dan 4 penghapusan
  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),
 	}
 	}
 }
 }