From 700dad1b8694aa629acea08a2e8f97f4c73a3f9b Mon Sep 17 00:00:00 2001 From: Molly Sophia Date: Thu, 1 Aug 2024 12:51:29 +0800 Subject: [PATCH] Fix build Signed-off-by: Molly Sophia --- src/llama.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llama.cpp b/src/llama.cpp index 83964eb2c..184b0870a 100644 --- a/src/llama.cpp +++ b/src/llama.cpp @@ -14811,7 +14811,7 @@ struct llm_build_context { } ggml_cgraph * build_rwkv() { - ggml_cgraph *gf = ggml_new_graph_custom(ctx0, LLAMA_MAX_NODES, false); + ggml_cgraph *gf = ggml_new_graph_custom(ctx0, llama_model_max_nodes(model), false); // Input embeddings, start of the model after tokenizing ({n_embd, n_tokens}) ggml_tensor * input_embeddings = llm_build_inp_embd(ctx0, lctx, hparams, batch, model.tok_embd, cb);