Skip to content

Commit 3d28b3b

Browse files
committed
graph : set res->params in llm_graph_context constructor
ggml-ci
1 parent c82a533 commit 3d28b3b

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

src/llama-graph.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -450,6 +450,7 @@ llm_graph_context::llm_graph_context(const llm_graph_params & params) :
450450
cb_func (params.cb),
451451
res (static_cast<llm_graph_result *>(params.res)),
452452
ctx0 (res->get_ctx()) {
453+
res->params = params;
453454
}
454455

455456
void llm_graph_context::cb(ggml_tensor * cur, const char * name, int il) const {

src/llama-model.cpp

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16479,9 +16479,6 @@ ggml_cgraph * llama_model::build_graph(const llm_graph_params & params) const {
1647916479
// add on pooling layer
1648016480
llm->build_pooling(gf, cls, cls_b, cls_out, cls_out_b);
1648116481

16482-
// TODO: updating the graph parameters here is a little bit obscure - figure out something better
16483-
llm->res->params = params;
16484-
1648516482
return llm->res->get_gf();
1648616483
}
1648716484

0 commit comments

Comments
 (0)