From 57570f9bd6b860ea0cc9064ceba4486be169a3f6 Mon Sep 17 00:00:00 2001
From: sid <sid@ragingfist.net>
Date: Mon, 9 Mar 2026 06:46:36 +0000
Subject: [PATCH] Read the persisted llama_kv_cell_ext for n_pos_per_embd > 1
 when doing state_read for all sequence ids

---
 src/llama-kv-cache.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/llama-kv-cache.cpp b/src/llama-kv-cache.cpp
index d80e8a70b..089d39d03 100644
--- a/src/llama-kv-cache.cpp
+++ b/src/llama-kv-cache.cpp
@@ -1955,6 +1955,12 @@ bool llama_kv_cache::state_read_meta(llama_io_read_i & io, uint32_t strm, uint32
 
             cells.pos_set(i, pos);
 
+            if (hparams.n_pos_per_embd() > 1) {
+                llama_kv_cell_ext ext;
+                io.read_to(&ext, sizeof(ext));
+                cells.ext_set(i, ext);
+            }
+
             for (uint32_t j = 0; j < n_seq_id; ++j) {
                 llama_seq_id seq_id;
                 io.read_to(&seq_id, sizeof(seq_id));