Skip to content

Commit ce77041

Browse files
committed
graph : set res->params in llm_graph_context constructor
ggml-ci
1 parent f7b99d7 commit ce77041

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
@@ -15879,9 +15879,6 @@ ggml_cgraph * llama_model::build_graph(const llm_graph_params & params) const {
1587915879
// add on pooling layer
1588015880
llm->build_pooling(gf, cls, cls_b, cls_out, cls_out_b);
1588115881

15882-
// TODO: updating the graph parameters here is a little bit obscure - figure out something better
15883-
llm->res->params = params;
15884-
1588515882
return llm->res->get_gf();
1588615883
}
1588715884

0 commit comments

Comments
 (0)