From: Shunta Saito <redacted>
Date: Sat, 25 Oct 2025 10:26:27 +0000 (+0900)
Subject: model : set res->t_embd in PLaMo2 models (#16766)
X-Git-Tag: upstream/0.0.7011~173
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=226f295f4dd92ad714533adc5497afed5fa88bb8;p=pkg%2Fggml%2Fsources%2Fllama.cpp

model : set res->t_embd in PLaMo2 models (#16766)
---

diff --git a/src/llama-model.cpp b/src/llama-model.cpp
index 7420a317..2a83d662 100644
--- a/src/llama-model.cpp
+++ b/src/llama-model.cpp
@@ -17965,6 +17965,8 @@ struct llm_build_plamo2 : public llm_graph_context_mamba {
         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);
         cb(cur, "result_output", -1);