Skip to content

Commit 5ef1722

Browse files
committed
fix for sd
1 parent 0954e9e commit 5ef1722

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

koboldcpp.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -559,6 +559,11 @@ def tryparseint(value):
559559
return int(value)
560560
except ValueError:
561561
return value
562+
def tryparsefloat(value):
563+
try:
564+
return float(value)
565+
except ValueError:
566+
return value
562567

563568
def is_incomplete_utf8_sequence(byte_seq): #note, this will only flag INCOMPLETE sequences, corrupted ones will be ignored.
564569
try:
@@ -1401,8 +1406,8 @@ def sd_generate(genparams):
14011406
prompt = forced_posprompt
14021407
init_images_arr = genparams.get("init_images", [])
14031408
init_images = ("" if (not init_images_arr or len(init_images_arr)==0 or not init_images_arr[0]) else init_images_arr[0])
1404-
denoising_strength = genparams.get("denoising_strength", 0.6)
1405-
cfg_scale = genparams.get("cfg_scale", 5)
1409+
denoising_strength = tryparsefloat(genparams.get("denoising_strength", 0.6))
1410+
cfg_scale = tryparsefloat(genparams.get("cfg_scale", 5))
14061411
sample_steps = tryparseint(genparams.get("steps", 20))
14071412
width = tryparseint(genparams.get("width", 512))
14081413
height = tryparseint(genparams.get("height", 512))

0 commit comments

Comments
 (0)