From fe2987d89447679b0481b21d025a118bb5c9eb28 Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Wed, 3 Nov 2021 12:42:04 -0400
Subject: [PATCH 1/2] Fix missing break statement in device_config

---
 aiserver.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/aiserver.py b/aiserver.py
index 29a9a3d1..e0097572 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -234,6 +234,7 @@ def device_config(model):
                 primaryselect = input("device ID> ")
                 if(primaryselect.isnumeric() and 0 <= int(primaryselect) < device_count):
                     breakmodel.primary_device = int(primaryselect)
+                    break
                 else:
                     print(f"{colors.RED}Please enter an integer between 0 and {device_count-1}.{colors.END}")
         else:

From 0fa47b12491890068987ca27b93dc783546c1f49 Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Wed, 3 Nov 2021 18:22:31 -0400
Subject: [PATCH 2/2] Fix budget calculation for stories with at least one
 non-prompt chunk

---
 aiserver.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index e0097572..5c421cd5 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -1255,7 +1255,7 @@ def calcsubmit(txt):
                 tokens = memtokens + witokens + prompttkns + tokens
             
             # Send completed bundle to generator
-            ln = len(tokens)
+            ln = len(tokens) + lnsp
             
             if(not vars.model in ["Colab", "OAI"]):
                 generate (