From 3084bfe633fce69ec6226c03d82185f12c85fbf3 Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Wed, 14 Jan 2026 12:39:08 +0200
Subject: [PATCH] llama : fix need_reserve when unsetting a sampler

---
 src/llama-context.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/llama-context.cpp b/src/llama-context.cpp
index 929678587e..c591c95be2 100644
--- a/src/llama-context.cpp
+++ b/src/llama-context.cpp
@@ -1057,6 +1057,8 @@ bool llama_context::set_sampler(llama_seq_id seq_id, llama_sampler * sampler) {
 
     sampling.samplers.erase(seq_id);
 
+    sched_need_reserve = true;
+
     return true;
 }