From 6b10a82c00064d4ead889b09d7fae9eff6927d57 Mon Sep 17 00:00:00 2001
From: sprayandwipe <spray.and.wipe@gmail.com>
Date: Sun, 15 Mar 2026 07:11:19 +0000
Subject: [PATCH] kv-cache : fix reading llama_kv_cell_ext during state read
 (#20273)

Co-authored-by: sid <sid@ragingfist.net>
---
 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 82fe58fac4..01166fac9c 100644
--- a/src/llama-kv-cache.cpp
+++ b/src/llama-kv-cache.cpp
@@ -1953,6 +1953,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));