From 4d5779d1bb1e85d6c205c5071049ff37aeaf017a Mon Sep 17 00:00:00 2001 From: Wagner Bruna Date: Thu, 3 Sep 2026 23:17:13 -0300 Subject: [PATCH] sd: allow changing flow_shift through the api (#2429) --- koboldcpp.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/koboldcpp.py b/koboldcpp.py index 4658b13b3..d2db55e9c 100644 --- a/koboldcpp.py +++ b/koboldcpp.py @@ -2940,7 +2940,10 @@ def sd_generate(genparams): cfg_scale = tryparsefloat(genparams.get("cfg_scale", 5),5) distilled_guidance = tryparsefloat(genparams.get("distilled_guidance", None), None) shifted_timestep = tryparseint(genparams.get("shifted_timestep", None), None) - flow_shift = tryparsefloat(genparams.get("flow_shift", None), None) + kcpp_params = genparams.get("kcpp_extra_args", {}) + kcpp_params = kcpp_params.get("params", {}) if isinstance(kcpp_params, dict) else {} + kcpp_params = kcpp_params if isinstance(kcpp_params, dict) else {} + flow_shift = tryparsefloat(kcpp_params.get("flow_shift", genparams.get("flow_shift", None)), None) sample_steps = tryparseint(genparams.get("steps", 20),20) width = tryparseint(genparams.get("width", 512),512) height = tryparseint(genparams.get("height", 512),512)