summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2023-03-11 12:44:21 +0200
committerGeorgi Gerganov <ggerganov@gmail.com>2023-03-11 12:45:01 +0200
commit7d9ed7b25fe17db3fc8848b5116d14682864ce8e (patch)
tree814ca85d3c35a5cd8b825fe9c21cf06f8748dd16
parent0c6803321c818f3f2da4a0693d20128b0f79ad28 (diff)
Bump memory buffer
-rw-r--r--main.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/main.cpp b/main.cpp
index d28fc916..2f474806 100644
--- a/main.cpp
+++ b/main.cpp
@@ -529,7 +529,7 @@ bool llama_eval(
const int d_key = n_embd/n_head;
- static size_t buf_size = 256u*1024*1024;
+ static size_t buf_size = 512u*1024*1024;
static void * buf = malloc(buf_size);
if (mem_per_token > 0 && mem_per_token*N > buf_size) {