model : set res->t_embd in SmallThinker models (#16782)
This commit is contained in:
@@ -19339,6 +19339,7 @@ struct llm_build_smallthinker : public llm_graph_context{
|
|||||||
|
|
||||||
cur = build_norm(cur, model.output_norm, NULL, LLM_NORM_RMS, -1);
|
cur = build_norm(cur, model.output_norm, NULL, LLM_NORM_RMS, -1);
|
||||||
cb(cur, "result_norm", -1);
|
cb(cur, "result_norm", -1);
|
||||||
|
res->t_embd = cur;
|
||||||
|
|
||||||
// lm_head
|
// lm_head
|
||||||
cur = build_lora_mm(model.output, cur);
|
cur = build_lora_mm(model.output, cur);
|
||||||
|
|||||||
Reference in New Issue
Block a user