Skip to content

Commit 65cd25d

Browse files
committed
some data sanitization
1 parent 06149f6 commit 65cd25d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

koboldcpp.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4311,7 +4311,7 @@ def guilaunch():
43114311

43124312
def export_vars():
43134313
nonlocal kcpp_exporting_template
4314-
args.threads = int(threads_var.get())
4314+
args.threads = (get_default_threads() if threads_var.get()=="" else int(threads_var.get()))
43154315
args.usemlock = usemlock.get() == 1
43164316
args.debugmode = debugmode.get()
43174317
args.launch = launchbrowser.get()==1
@@ -4363,7 +4363,7 @@ def export_vars():
43634363
if runopts_var.get() == "Use Vulkan (Old CPU)":
43644364
args.noavx2 = True
43654365
if gpulayers_var.get():
4366-
args.gpulayers = int(gpulayers_var.get())
4366+
args.gpulayers = (0 if gpulayers_var.get()=="" else int(gpulayers_var.get()))
43674367
if runopts_var.get()=="Use CPU":
43684368
args.usecpu = True
43694369
if runopts_var.get()=="Use CPU (Old CPU)":
@@ -4480,7 +4480,7 @@ def export_vars():
44804480
args.ttsmodel = tts_model_var.get()
44814481
args.ttswavtokenizer = wavtokenizer_var.get()
44824482
args.ttsgpu = (ttsgpu_var.get()==1)
4483-
args.ttsmaxlen = int(ttsmaxlen_var.get())
4483+
args.ttsmaxlen = (default_ttsmaxlen if ttsmaxlen_var.get()=="" else int(ttsmaxlen_var.get()))
44844484

44854485
args.admin = (admin_var.get()==1 and not args.cli)
44864486
args.admindir = admin_dir_var.get()

0 commit comments

Comments
 (0)