From 4d38795563f933b1a3f4ca6956d9aabeb445857e Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Mon, 1 May 2023 12:10:21 +0800
Subject: [PATCH] add UI for token unbanning

---
 koboldcpp.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/koboldcpp.py b/koboldcpp.py
index 354d19c13a8ef..52d3b5ffa201c 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -181,7 +181,7 @@ def generate(prompt,max_length=20, max_context_length=512,temperature=0.8,top_k=
 maxlen = 128
 modelbusy = False
 defaultport = 5001
-KcppVersion = "1.16"
+KcppVersion = "1.17"
 
 class ServerRequestHandler(http.server.SimpleHTTPRequestHandler):
     sys_version = ""
@@ -432,8 +432,10 @@ def guilaunch():
         stream = tk.IntVar()
         smartcontext = tk.IntVar()
         launchbrowser = tk.IntVar(value=1)
+        unbantokens = tk.IntVar()
         tk.Checkbutton(root, text='Streaming Mode',variable=stream, onvalue=1, offvalue=0).pack()
         tk.Checkbutton(root, text='Use SmartContext',variable=smartcontext, onvalue=1, offvalue=0).pack()
+        tk.Checkbutton(root, text='Unban Tokens',variable=unbantokens, onvalue=1, offvalue=0).pack()
         tk.Checkbutton(root, text='Launch Browser',variable=launchbrowser, onvalue=1, offvalue=0).pack()
     
         # Create button, it will change label text
@@ -452,6 +454,7 @@ def guilaunch():
         args.stream = (stream.get()==1)
         args.smartcontext = (smartcontext.get()==1)
         args.launch = (launchbrowser.get()==1)
+        args.unbantokens = (unbantokens.get()==1)
         selchoice = runchoice.get()
         
         if selchoice==opts[1]: