File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff 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
563568def 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 ))
You can’t perform that action at this time.
0 commit comments