Skip to content

Commit 7cce4f8

Browse files
authored
model : set res->t_embd in SmallThinker models (#16782)
1 parent 8d88628 commit 7cce4f8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/llama-model.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19339,6 +19339,7 @@ struct llm_build_smallthinker : public llm_graph_context{
1933919339

1934019340
cur = build_norm(cur, model.output_norm, NULL, LLM_NORM_RMS, -1);
1934119341
cb(cur, "result_norm", -1);
19342+
res->t_embd = cur;
1934219343

1934319344
// lm_head
1934419345
cur = build_lora_mm(model.output, cur);

0 commit comments

Comments
 (0)