From 2a07f2dc2c0ec1b18cbaab65dcb4f8619b6762ca Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Fri, 1 Nov 2024 22:42:57 +0800
Subject: [PATCH] minor fix

---
 klite.embd   | 2 +-
 koboldcpp.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/klite.embd b/klite.embd
index ab7faf382..12c4f23f1 100644
--- a/klite.embd
+++ b/klite.embd
@@ -19254,7 +19254,7 @@ Current version indicated by LITEVER below.
 							</div>
 
 							<div class="settinglabel settingcell">
-								<div class="justifyleft settingsmall">DRY (If supported) <span class="helpicon">?<span class="helptext">An advanced multi-token repetition penalty. May not be available depending on backend, not supported on Horde.</span></span></div>
+								<div class="justifyleft settingsmall">DRY (If supported) <span class="helpicon">?<span class="helptext">An advanced multi-token repetition penalty. Range controlled by Rep-Pen Range. May not be available depending on backend, not supported on Horde.</span></span></div>
 								<div id="drysupporteddiv">
 								<div style="display:flex">
 									<div class="settinglabel settingcell">
diff --git a/koboldcpp.py b/koboldcpp.py
index 65c881e41..9ead16b10 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -1084,7 +1084,7 @@ def generate(genparams, is_quiet=False, stream_flag=False):
     if pendingabortkey!="" and pendingabortkey==genkey:
         print(f"\nDeferred Abort for GenKey: {pendingabortkey}")
         pendingabortkey = ""
-        return {"text":"","status":-1,"stopreason":-1, "prompt_tokens":0, "completion_tokens": 0}
+        return {"text":"","status":-1,"stopreason":-1, "prompt_tokens":0, "completion_tokens": 0, "total_tokens": 0}
     else:
         ret = handle.generate(inputs)
         outstr = ""
@@ -1502,7 +1502,7 @@ class ServerRequestHandler(http.server.SimpleHTTPRequestHandler):
 
             return generate(genparams=genparams,is_quiet=is_quiet,stream_flag=stream_flag)
 
-        genout = {"text": "", "status": -1, "stopreason": -1, "prompt_tokens":0, "completion_tokens": 0}
+        genout = {"text": "", "status": -1, "stopreason": -1, "prompt_tokens":0, "completion_tokens": 0, "total_tokens": 0}
         if stream_flag:
             loop = asyncio.get_event_loop()
             executor = ThreadPoolExecutor()