Temporary bump the memory buffer size - hopefully fix issues from 483bab2e

pull/473/head^2 master-31572d9
Georgi Gerganov 1 year ago
parent f4f5362edb
commit 31572d9665
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

@ -632,7 +632,7 @@ static bool llama_eval_internal(
auto & mem_per_token = lctx.mem_per_token;
// TODO: fix this hardcoded size
static size_t buf_size = 512u*1024*1024;
static size_t buf_size = 2048u*1024*1024; // TMP !!!
static void * buf = malloc(buf_size);
if (mem_per_token > 0 && mem_per_token*N > buf_size) {

Loading…
Cancel
Save