Skip to content

Commit 226f295

Browse files
authored
model : set res->t_embd in PLaMo2 models (ggml-org#16766)
1 parent f90b4a8 commit 226f295

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/llama-model.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17965,6 +17965,8 @@ struct llm_build_plamo2 : public llm_graph_context_mamba {
1796517965
cur = build_norm(cur, model.output_norm, NULL, LLM_NORM_RMS, -1);
1796617966
cb(cur, "result_norm", -1);
1796717967

17968+
res->t_embd = cur;
17969+
1796817970
// lm_head
1796917971
cur = build_lora_mm(model.output, cur);
1797017972
cb(cur, "result_output", -1);

0 commit comments

Comments
 (0)