Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-20 01:31:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

sd: make aborting on client disconnect opt-out (#2322)

Browse Source
This commit is contained in:
Wagner Bruna
2026-07-11 11:21:22 -03:00
committed by GitHub
parent 0240b72895
commit ce3623ad8b
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
koboldcpp.py
+5 -1
View File
@@ -7510,7 +7510,11 @@ Change Mode<br>
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"]