graph : set res->params in llm_graph_context constructor

ggml-ci

Author: ggerganov

src/llama-graph.cpp

@@ -450,6 +450,7 @@ llm_graph_context::llm_graph_context(const llm_graph_params & params) :
     cb_func          (params.cb),
     res              (static_cast<llm_graph_result *>(params.res)),
     ctx0             (res->get_ctx()) {
+        res->params = params;
     }
 
 void llm_graph_context::cb(ggml_tensor * cur, const char * name, int il) const {

src/llama-model.cpp

@@ -16268,9 +16268,6 @@ ggml_cgraph * llama_model::build_graph(const llm_graph_params & params) const {
     // add on pooling layer
     llm->build_pooling(gf, cls, cls_b, cls_out, cls_out_b);
 
-    // TODO: updating the graph parameters here is a little bit obscure - figure out something better
-    llm->res->params = params;
-
     return llm->res->get_gf();
 }