From efaa30ef1118831c9f27e3341336f83e271cae1a Mon Sep 17 00:00:00 2001 From: maekawatoshiki Date: Mon, 13 Mar 2023 10:38:41 +0900 Subject: [PATCH] Use vector --- main.cpp | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/main.cpp b/main.cpp index b5eae67..5b70937 100644 --- a/main.cpp +++ b/main.cpp @@ -74,10 +74,10 @@ bool llama_model_load(const std::string & fname, llama_model & model, gpt_vocab printf("%s: loading model from '%s' - please wait ...\n", __func__, fname.c_str()); const size_t f_buf_size = 1024*1024; - char *f_buf = (char *)malloc(f_buf_size); + std::vector f_buf(f_buf_size); auto fin = std::ifstream(fname, std::ios::binary); - fin.rdbuf()->pubsetbuf(f_buf, f_buf_size); + fin.rdbuf()->pubsetbuf(f_buf.data(), f_buf_size); if (!fin) { fprintf(stderr, "%s: failed to open '%s'\n", __func__, fname.c_str()); return false; @@ -315,7 +315,7 @@ bool llama_model_load(const std::string & fname, llama_model & model, gpt_vocab printf("%s: loading model part %d/%d from '%s'\n", __func__, i+1, n_parts, fname_part.c_str()); fin = std::ifstream(fname_part, std::ios::binary); - fin.rdbuf()->pubsetbuf(f_buf, f_buf_size); + fin.rdbuf()->pubsetbuf(f_buf.data(), f_buf_size); fin.seekg(file_offset); // load weights @@ -501,8 +501,6 @@ bool llama_model_load(const std::string & fname, llama_model & model, gpt_vocab fin.close(); } - free(f_buf); - return true; }