fix: crash when calling `llama_state_get_size` on a context without a KV cache (...

diff --git a/src/llama-context.cpp b/src/llama-context.cpp
index 62246c10dab089f4583f0174492792eb27e7c543..1b76317da0ec9f81a5e637b723b75150b777f977 100644 (file)
--- a/src/llama-context.cpp
+++ b/src/llama-context.cpp
@@ -1704,10 +1704,12 @@ size_t llama_context::state_write_data(llama_io_write_i & io) {
         }
     }
 
-    LLAMA_LOG_DEBUG("%s: - writing KV self\n", __func__);
     llama_kv_cache * kv_self = static_cast<llama_kv_cache *>(memory.get());
-
-    kv_self->state_write(io);
+    
+    if (kv_self != nullptr) {
+        LLAMA_LOG_DEBUG("%s: - writing KV self\n", __func__);
+        kv_self->state_write(io);
+    }
 
     return io.n_bytes();
 }