vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2025-09-11 01:24:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for sd

Browse source
This commit is contained in:
Concedo 2025-03-16 17:02:42 +08:00
parent 0954e9e476
commit 5ef1722d5f
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
koboldcpp.py
View file

@ -559,6 +559,11 @@ def tryparseint(value):
return int(value) return int(value)
except ValueError: except ValueError:
return value return value
def tryparsefloat(value):
try:
return float(value)
except ValueError:
return value
def is_incomplete_utf8_sequence(byte_seq): #note, this will only flag INCOMPLETE sequences, corrupted ones will be ignored. def is_incomplete_utf8_sequence(byte_seq): #note, this will only flag INCOMPLETE sequences, corrupted ones will be ignored.
try: try:
@ -1401,8 +1406,8 @@ def sd_generate(genparams):
prompt = forced_posprompt prompt = forced_posprompt
init_images_arr = genparams.get("init_images", []) init_images_arr = genparams.get("init_images", [])
init_images = ("" if (not init_images_arr or len(init_images_arr)==0 or not init_images_arr[0]) else init_images_arr[0]) init_images = ("" if (not init_images_arr or len(init_images_arr)==0 or not init_images_arr[0]) else init_images_arr[0])
denoising_strength = genparams.get("denoising_strength", 0.6) denoising_strength = tryparsefloat(genparams.get("denoising_strength", 0.6))
cfg_scale = genparams.get("cfg_scale", 5) cfg_scale = tryparsefloat(genparams.get("cfg_scale", 5))
sample_steps = tryparseint(genparams.get("steps", 20)) sample_steps = tryparseint(genparams.get("steps", 20))
width = tryparseint(genparams.get("width", 512)) width = tryparseint(genparams.get("width", 512))
height = tryparseint(genparams.get("height", 512)) height = tryparseint(genparams.get("height", 512))