From f2f9d9d2b3fa6a562510c50739224e75004c861d Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Sun, 12 Jul 2026 22:37:12 +0800
Subject: [PATCH] cont: abort bug for images

---
 koboldcpp.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/koboldcpp.py b/koboldcpp.py
index a4e05cad0..c4714125e 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -7515,7 +7515,7 @@ Change Mode<br>
                                 genparams['lora'] = lora_map_name_to_path(loras)
                         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:
+                        if override_abort_gen is not None and tryparseint(override_abort_gen, 1):
                             abort_gen = None
                         gen = asyncio.run(self.handle_image_request(sd_generate, genparams, abort_gen))
                         gendat = gen["data"]