Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-18 16:55:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

sd: allow changing flow_shift through the api (#2429)

Browse Source
This commit is contained in:
Wagner Bruna
2026-09-03 23:17:13 -03:00
committed by GitHub
parent 5ac1407e87
commit 4d5779d1bb
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
koboldcpp.py
+4 -1
View File
@@ -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)