model : set res->t_embd in SmallThinker models (#16782)

This commit is contained in:
Sigbjørn Skjæret 2025-10-26 16:08:52 +01:00 committed by GitHub
parent 8d8862829c
commit 7cce4f8158
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions

View File

@ -19339,6 +19339,7 @@ struct llm_build_smallthinker : public llm_graph_context{
cur = build_norm(cur, model.output_norm, NULL, LLM_NORM_RMS, -1);
cb(cur, "result_norm", -1);
res->t_embd = cur;
// lm_head
cur = build_lora_mm(model.output, cur);