From ce3623ad8b1a616033930a23ec9a9796792c67ae Mon Sep 17 00:00:00 2001 From: Wagner Bruna Date: Sat, 11 Jul 2026 11:21:22 -0300 Subject: [PATCH] sd: make aborting on client disconnect opt-out (#2322) --- koboldcpp.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/koboldcpp.py b/koboldcpp.py index dacbd9d24..40642af61 100644 --- a/koboldcpp.py +++ b/koboldcpp.py @@ -7510,7 +7510,11 @@ Change Mode
if loras: genparams['prompt'] = prompt genparams['lora'] = lora_map_name_to_path(loras) - gen = asyncio.run(self.handle_image_request(sd_generate, genparams, handle.sd_abort_generation)) + abort_gen = handle.sd_abort_generation + override_abort_gen = genparams.get('kcpp_extra_args', {}).get('keep_image_gen_on_disconnect', gendefaults.get('keep_image_gen_on_disconnect')) + if override_abort_gen is not None and not tryparseint(override_abort_gen, 1): + abort_gen = None + gen = asyncio.run(self.handle_image_request(sd_generate, genparams, abort_gen)) gendat = gen["data"] genanim = gen["animated"] gendatextra = gen["data_extra"]