From 468f8f392ee035ab6dedbbbfcfc89982634f570d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sigbj=C3=B8rn=20Skj=C3=A6ret?= <sigbjorn.skjaeret@scala.com>
Date: Sun, 26 Oct 2025 15:00:36 +0100
Subject: [PATCH] set res->t_embd in SmallThinker models

---
 src/llama-model.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/llama-model.cpp b/src/llama-model.cpp
index 2a83d66279b79..a6a6fa7d7198a 100644
--- a/src/llama-model.cpp
+++ b/src/llama-model.cpp
@@ -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);